Searched refs:shape_expected (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 1336 std::vector<int64> shape_expected = {10, 20, 30, 40}; in TEST_F() local
1337 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
1341 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1345 std::vector<int64> shape_expected = {40, 20, 90, 40}; in TEST_F() local
1346 GetTensorProto(DT_INT64, {4}, shape_expected, in TEST_F()
1350 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1354 std::vector<int64> shape_expected = {10, 20, 30, 40}; in TEST_F() local
1355 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
1359 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
1363 std::vector<int64> shape_expected = {40, 20, 90, 40}; in TEST_F() local
[all …]
|