Searched refs:InferGetTupleElementShape (Results 1 – 6 of 6) sorted by relevance
200 static StatusOr<Shape> InferGetTupleElementShape(const Shape& arg,
193 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in MakeGetTupleElementHlo()
890 ShapeInference::InferGetTupleElementShape(tuple_shape, 0); in TEST_F()892 ShapeInference::InferGetTupleElementShape(tuple_shape, 1); in TEST_F()902 ShapeInference::InferGetTupleElementShape(tuple_shape, -1); in TEST_F()904 ShapeInference::InferGetTupleElementShape(tuple_shape, 2); in TEST_F()
418 ShapeInference::InferGetTupleElementShape( in HandleGetTupleElement()
1184 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in HandleGetTupleElement()
2501 /* static */ StatusOr<Shape> ShapeInference::InferGetTupleElementShape( in InferGetTupleElementShape() function in xla::ShapeInference