Searched refs:AddTestWeights (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1215 void AddTestWeights(const char* name, const std::vector<int>& dims, in AddTestWeights() function in tensorflow::tensorrt::convert::OpConverterTest
1456 AddTestWeights<int32>("weights", {4}, {1, 0, 2, 3}); in TEST_F()
1464 AddTestWeights<int32>("weights", {3}, {0, 1, 2}); in TEST_F()
1473 AddTestWeights<int32>("weights", {4}, {0, 3, 1, 2}); in TEST_F()
1518 AddTestWeights<int32>("weights", {0}, {}); in TEST_F()
1543 AddTestWeights<int32>("weights", {4}, params[i].shape); in TEST_F()
1560 AddTestWeights<int32>("weights", {4}, ok_params[i].shape); in TEST_F()
1603 AddTestWeights<int32>("weights", {2, 1}, {3, 5}); in TEST_F()
1614 AddTestWeights<float>("weights", {2, 2}, {0, 1, 2, 3}); in TEST_F()
1625 AddTestWeights<float>("weights", {2, 2}, {0, 1, 2, 3}); in TEST_F()
[all …]
|