Searched refs:InferGetTupleElementShape (Results 1 – 6 of 6) sorted by relevance
219 static StatusOr<Shape> InferGetTupleElementShape(const Shape& arg,
228 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in MakeGetTupleElementHlo()
1337 ShapeInference::InferGetTupleElementShape(tuple_shape, 0); in TEST_F()1339 ShapeInference::InferGetTupleElementShape(tuple_shape, 1); in TEST_F()1349 ShapeInference::InferGetTupleElementShape(tuple_shape, -1); in TEST_F()1351 ShapeInference::InferGetTupleElementShape(tuple_shape, 2); in TEST_F()
636 ShapeInference::InferGetTupleElementShape( in HandleGetTupleElement()
1858 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in HandleGetTupleElement()
2728 /* static */ StatusOr<Shape> ShapeInference::InferGetTupleElementShape( in InferGetTupleElementShape() function in xla::ShapeInference