Home
last modified time | relevance | path

Searched refs:InferGetTupleElementShape (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.h250 static StatusOr<Shape> InferGetTupleElementShape(const Shape& arg,
H A Dshape_inference_test.cc1372 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()
H A Dhlo_creation_utils.cc266 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in MakeGetTupleElementHlo()
H A Dhlo_verifier.cc999 ShapeInference::InferGetTupleElementShape( in HandleGetTupleElement()
H A Dshape_inference.cc2845 /* static */ StatusOr<Shape> ShapeInference::InferGetTupleElementShape( in InferGetTupleElementShape() function in xla::ShapeInference
H A Dhlo_evaluator.cc3013 ShapeInference::InferGetTupleElementShape(operand->shape(), index)); in HandleGetTupleElement()