Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc2211 ShapeInference::InferScatterShape( in TEST_F()
2225 ShapeInference::InferScatterShape( in TEST_F()
2239 ShapeInference::InferScatterShape( in TEST_F()
2253 ShapeInference::InferScatterShape( in TEST_F()
2266 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
2283 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
2301 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
2320 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
2340 ShapeInference::InferScatterShape( in TEST_F()
2355 ShapeInference::InferScatterShape( in TEST_F()
[all …]
Dshape_inference.h290 static StatusOr<Shape> InferScatterShape(
Dhlo_verifier.cc820 scatter, ShapeInference::InferScatterShape( in HandleScatter()
Dshape_inference.cc3163 /*static*/ StatusOr<Shape> ShapeInference::InferScatterShape( in InferScatterShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1878 ShapeInference::InferScatterShape( in Scatter()