Searched refs:input_is_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 938 const std::vector<int>& expected_tensor_dims, bool input_is_tensor, in TestPrepareTensorForShape() argument
943 if (input_is_tensor) { in TestPrepareTensorForShape()
972 for (bool input_is_tensor : {true, false}) { in TEST_F()
975 TestPrepareTensorForShape({2, 3, 5}, {2, 3, 6}, {}, input_is_tensor, in TEST_F()
981 TestPrepareTensorForShape({2, 3, 5}, {10, 3}, {10, 3}, input_is_tensor, in TEST_F()
986 TestPrepareTensorForShape({1, 1}, {}, {}, input_is_tensor, converter_.get(), in TEST_F()
|