Searched refs:ExpectTensorShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 1304 void ExpectTensorShape(const std::vector<int64>& expected, in ExpectTensorShape() function
1341 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1350 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1359 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1368 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1649 ExpectTensorShape({1, 1, 1, 1}, y); in TEST_F()
1658 ExpectTensorShape({1}, y); in TEST_F()
1667 ExpectTensorShape({1, 1}, y); in TEST_F()
1678 ExpectTensorShape({10, 20}, y); in TEST_F()
1684 ExpectTensorShape({10, 20, 1, 1}, z); in TEST_F()
[all …]
|