Searched refs:AddTestTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1560 void AddTestTensor( in AddTestTensor() function in tensorflow::tensorrt::convert::OpConverterTest
1818 void AddTestTensor(const string& name, const std::vector<int32>& dims, in AddTestTensor() function in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
1859 void AddTestTensor(const string& name, const std::vector<int32>& dims, in AddTestTensor() function in tensorflow::tensorrt::convert::ParameterizedOpConverterTestBase
1862 AddTestTensor<T>(name, dims, tf_type_, values, partial_input_shape_dims);
2246 AddTestTensor(node_input[i].name, node_input[i].dims, tf_type_, in TEST_P()
2308 AddTestTensor("input", p.input_dims, {1, 2, 3, 4, 5, 6}, in TEST_P()
2311 AddTestTensor("weights", {3}); in TEST_P()
2332 AddTestTensor("input", {1, 2, 3}); in TEST_F()
2333 AddTestTensor("weights", {3}); in TEST_F()
2341 AddTestTensor("input", {1, 2, 3}); in TEST_F()
[all …]
|