Searched refs:outfeed_operand (Results 1 – 1 of 1) sorted by relevance
2985 HloInstruction* outfeed_operand) -> HloInstruction* { in HandleOutfeed() argument2999 ShapeUtil::GetSubshape(element_shape, {i}), outfeed_operand, in HandleOutfeed()3009 return outfeed_operand; in HandleOutfeed()3017 slice_shape, outfeed_operand, start_indices, in HandleOutfeed()3022 return outfeed_operand; in HandleOutfeed()