Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc2913 ShapeInference::InferScatterShape( in TEST_F()
2927 ShapeInference::InferScatterShape( in TEST_F()
2941 ShapeInference::InferScatterShape( in TEST_F()
2955 ShapeInference::InferScatterShape( in TEST_F()
2968 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
2985 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
3003 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
3022 StatusOr<Shape> statusor = ShapeInference::InferScatterShape( in TEST_F()
3042 ShapeInference::InferScatterShape( in TEST_F()
3057 ShapeInference::InferScatterShape( in TEST_F()
[all …]
Dshape_inference.h321 static StatusOr<Shape> InferScatterShape(
Dhlo_verifier.cc1141 scatter, ShapeInference::InferScatterShape( in HandleScatter()
Dshape_inference.cc3538 /*static*/ StatusOr<Shape> ShapeInference::InferScatterShape( in InferScatterShape() function in xla::ShapeInference
Dhlo_parser.cc2524 return ShapeInference::InferScatterShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2314 Shape shape, ShapeInference::InferScatterShape( in Scatter()