Searched refs:ExpectTensorShape (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 925 void ExpectTensorShape(const std::vector<int64>& expected, in ExpectTensorShape() function
964 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
973 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
982 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
991 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1260 ExpectTensorShape({1, 1, 1, 1}, y); in TEST_F()
1269 ExpectTensorShape({1}, y); in TEST_F()
1278 ExpectTensorShape({1, 1}, y); in TEST_F()
1289 ExpectTensorShape({10, 20}, y); in TEST_F()
1295 ExpectTensorShape({10, 20, 1, 1}, z); in TEST_F()
[all …]
|