Searched refs:AddTestWeights (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1574 void AddTestWeights(const string& name, const std::vector<int>& dims, in AddTestWeights() function in tensorflow::tensorrt::convert::OpConverterTest
1601 void AddTestWeights(const string& name, const std::vector<int>& dims, in AddTestWeights() function in tensorflow::tensorrt::convert::OpConverterTest
1604 AddTestWeights(name, dims, CastTestVector<T, float>(values)); in AddTestWeights()
1606 AddTestWeights(name, dims, CastTestVector<T, Eigen::half>(values)); in AddTestWeights()
1608 AddTestWeights(name, dims, CastTestVector<T, int32>(values)); in AddTestWeights()
2251 AddTestWeights(node_input[i].name, node_input[i].dims, in TEST_P()
2313 AddTestWeights<int32>("weights", {static_cast<int>(p.param.size())}, in TEST_P()
2342 AddTestWeights<int32>("weights", {0}, {}); in TEST_F()
2351 AddTestWeights<int32>("weights", {1, 0, 1}, {}); in TEST_F()
2375 AddTestWeights<int32>("weights", {4}, params[i].shape); in TEST_F()
[all …]
|