Home
last modified time | relevance | path

Searched refs:InferGatherShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc1778 ShapeInference::InferGatherShape( in TEST_F()
1793 ShapeInference::InferGatherShape( in TEST_F()
1808 ShapeInference::InferGatherShape( in TEST_F()
1824 ShapeInference::InferGatherShape( in TEST_F()
1841 ShapeInference::InferGatherShape( in TEST_F()
1859 ShapeInference::InferGatherShape( in TEST_F()
1877 ShapeInference::InferGatherShape( in TEST_F()
1895 ShapeInference::InferGatherShape( in TEST_F()
1910 StatusOr<Shape> statusor = ShapeInference::InferGatherShape( in TEST_F()
1925 StatusOr<Shape> statusor = ShapeInference::InferGatherShape( in TEST_F()
[all …]
Dshape_inference.h282 static StatusOr<Shape> InferGatherShape(
Dhlo_verifier.cc813 ShapeInference::InferGatherShape( in HandleGather()
Dshape_inference.cc2957 /*static*/ StatusOr<Shape> ShapeInference::InferGatherShape( in InferGatherShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1849 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferGatherShape( in Gather()