Searched refs:outfeed_data (Results 1 – 1 of 1) sorted by relevance
2977 auto outfeed_data = branch_b.AddInstruction( in HandleOutfeed() local2988 ShapeUtil::GetSubshape(outfeed_data->shape(), index); in HandleOutfeed()3024 outfeed_data = slice_outfeed({}, outfeed_data); in HandleOutfeed()3029 per_branch_partitioned_shapes[i], outfeed_data, outfeed_token, in HandleOutfeed()