Searched refs:AddTestTensorWithTFDims (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 1343 void AddTestTensorWithTFDims( in AddTestTensorWithTFDims() function in tensorflow::tensorrt::convert::OpConverterTest
1389 AddTestTensorWithTFDims(name, dims_vec, trt_dtype); in AddTestTensor()
1853 AddTestTensorWithTFDims(name, partial_shape, trt_type, add_input_status);
6145 AddTestTensorWithTFDims("input", {-1, -1, -1, -1}, trt_type); in TEST_P()
|