Searched refs:AddTestTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1194 void AddTestTensor( in AddTestTensor() function in tensorflow::tensorrt::convert::OpConverterTest
1410 AddTestTensor("input", {1}); in TEST_F()
1446 AddTestTensor("input", {1, 2, 3}); in TEST_F()
1447 AddTestTensor("weights", {3}); in TEST_F()
1455 AddTestTensor("input", {1, 2, 3}); in TEST_F()
1463 AddTestTensor("input", {1, 2, 3}); in TEST_F()
1472 AddTestTensor("input", {1, 2, 3}); in TEST_F()
1508 AddTestTensor("input", {1, 2, 3}); in TEST_F()
1509 AddTestTensor("weights", {3}); in TEST_F()
1517 AddTestTensor("input", {1, 2, 3}); in TEST_F()
[all …]
|