Searched refs:InferGetTupleElementShape (Results 1 – 6 of 6) sorted by relevance
250 static StatusOr<Shape> InferGetTupleElementShape(const Shape& arg,
1372 ShapeInference::InferGetTupleElementShape(tuple_shape, 0); in TEST_F()1374 ShapeInference::InferGetTupleElementShape(tuple_shape, 1); in TEST_F()1384 ShapeInference::InferGetTupleElementShape(tuple_shape, -1); in TEST_F()1386 ShapeInference::InferGetTupleElementShape(tuple_shape, 2); in TEST_F()
266 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in MakeGetTupleElementHlo()
999 ShapeInference::InferGetTupleElementShape( in HandleGetTupleElement()
2845 /* static */ StatusOr<Shape> ShapeInference::InferGetTupleElementShape( in InferGetTupleElementShape() function in xla::ShapeInference
3013 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in HandleGetTupleElement()