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