Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc2427 ShapeInference::InferGatherShape( in TEST_F()
2442 ShapeInference::InferGatherShape( in TEST_F()
2457 ShapeInference::InferGatherShape( in TEST_F()
2473 ShapeInference::InferGatherShape( in TEST_F()
2490 ShapeInference::InferGatherShape( in TEST_F()
2507 ShapeInference::InferGatherShape( in TEST_F()
2524 ShapeInference::InferGatherShape( in TEST_F()
2542 ShapeInference::InferGatherShape( in TEST_F()
2560 ShapeInference::InferGatherShape( in TEST_F()
2578 ShapeInference::InferGatherShape( in TEST_F()
[all …]
Dshape_inference.h313 static StatusOr<Shape> InferGatherShape(
Dhlo_verifier.cc1134 ShapeInference::InferGatherShape( in HandleGather()
Dshape_inference.cc3302 /*static*/ StatusOr<Shape> ShapeInference::InferGatherShape( in InferGatherShape() function in xla::ShapeInference
Dhlo_parser.cc2473 return ShapeInference::InferGatherShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc148 ShapeInference::InferGatherShape(replicated_operand.hlo()->shape(), in PartitionIndexOnlyPartition()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2278 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferGatherShape( in Gather()