Searched refs:TestBinaryTensorOpWeightFallback (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1889 void TestBinaryTensorOpWeightFallback(OpConverterTest* test, in TestBinaryTensorOpWeightFallback() function
2058 TestBinaryTensorOpWeightFallback<ops::Minimum>(this, {1, 1, 1}, {1}); in TEST_F()
2060 TestBinaryTensorOpWeightFallback<ops::Add>(this, {1, 1}, {1}); in TEST_F()
2062 TestBinaryTensorOpWeightFallback<ops::Add>( in TEST_F()
2067 TestBinaryTensorOpWeightFallback<ops::Add>(this, {1, 1, 1}, {1, 2, 1}); in TEST_F()
2069 TestBinaryTensorOpWeightFallback<ops::Add>(this, {1, 2, 1}, {2}); in TEST_F()
|