Searched defs:tuple_element_shape (Results 1 – 3 of 3) sorted by relevance
203 const Shape& tuple_element_shape = ShapeUtil::GetSubshape(shape, {i}); in TransferLiteralToInfeedOnCpu() local251 const Shape& tuple_element_shape = in TransferLiteralFromOutfeedOnCpu() local
422 const Shape& tuple_element_shape = in HandleInfeed() local533 const Shape& tuple_element_shape = in HandleOutfeed() local
1410 for (const Shape& tuple_element_shape : shape().tuple_shapes()) { in output_shapes() local