Searched refs:TestPrepareTensorForShape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 728 void TestPrepareTensorForShape( in TestPrepareTensorForShape() function
770 TestPrepareTensorForShape({2, 3, 5}, {2, 3, 6}, {}, input_is_tensor, in TEST_F()
776 TestPrepareTensorForShape({2, 3, 5}, {10, 3}, {10, 3}, input_is_tensor, in TEST_F()
781 TestPrepareTensorForShape({1, 1}, {}, {}, input_is_tensor, converter_.get(), in TEST_F()
787 TestPrepareTensorForShape({}, {1, 1}, {1, 1}, /*input_is_tensor=*/true, in TEST_F()
795 TestPrepareTensorForShape({2, 3, 5}, {-1, 2}, {15, 2}, in TEST_F()
801 TestPrepareTensorForShape({2, 3, 5}, {-1, 2}, {15, 2}, in TEST_F()
|