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()
|