Searched refs:InferReshapeShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 248 static StatusOr<Shape> InferReshapeShape(const Shape& operand,
|
| D | shape_inference_test.cc | 1384 auto status = ShapeInference::InferReshapeShape(operand, {1, 0}, {1}, in TEST_F()
1396 auto status = ShapeInference::InferReshapeShape(operand, {0}, {1, 10}, in TEST_F()
1407 auto status = ShapeInference::InferReshapeShape(operand, {1, 0}, {60}, in TEST_F()
1417 auto status = ShapeInference::InferReshapeShape(operand, {1, 0}, {2, 3, 10}, in TEST_F()
|
| D | shape_inference.cc | 2943 /* static */ StatusOr<Shape> ShapeInference::InferReshapeShape( in InferReshapeShape() function in xla::ShapeInference
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1120 TF_ASSIGN_OR_RETURN(const Shape shape, ShapeInference::InferReshapeShape( in Reshape()
|