Searched defs:tuple_element_shape (Results 1 – 2 of 2) sorted by relevance
122 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i}); in TransferLiteralToInfeed() local207 const Shape& tuple_element_shape = in TransferLiteralFromOutfeed() local
358 const Shape& tuple_element_shape = in HandleInfeed() local492 const Shape& tuple_element_shape = in HandleOutfeed() local