Searched refs:infeed_element (Results 1 – 1 of 1) sorted by relevance
2575 HloInstruction* infeed_element) -> HloInstruction* { in HandleInfeed() argument2578 return infeed_element; in HandleInfeed()2591 ShapeUtil::GetSubshape(element_shape, {i}), infeed_element, in HandleInfeed()2600 return infeed_element; in HandleInfeed()2604 return PadToShape(infeed_element, pad_shape, &branch_b); in HandleInfeed()