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 | 959 std::vector<int64> shape_expected = {10, 20, 30, 40}; in TEST_F() local
960 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
964 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
968 std::vector<int64> shape_expected = {40, 20, 90, 40}; in TEST_F() local
969 GetTensorProto(DT_INT64, {4}, shape_expected, in TEST_F()
973 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
977 std::vector<int64> shape_expected = {10, 20, 30, 40}; in TEST_F() local
978 GetTensorProto(DT_INT32, {4}, shape_expected, in TEST_F()
982 ExpectTensorShape(shape_expected, tensor_shape_proto); in TEST_F()
986 std::vector<int64> shape_expected = {40, 20, 90, 40}; in TEST_F() local
[all …]
|