Searched refs:input_is_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 730 const std::vector<int>& expected_tensor_dims, bool input_is_tensor, in TestPrepareTensorForShape() argument
735 if (input_is_tensor) { in TestPrepareTensorForShape()
767 for (bool input_is_tensor : {true, false}) { in TEST_F()
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()
2804 for (bool input_is_tensor : {true, false}) { in TEST_P()
2844 if (input_is_tensor) { in TEST_P()
4336 for (bool input_is_tensor : {true, false}) { in TEST_P()
4344 if (input_is_tensor) { in TEST_P()
|