Searched refs:AddTestTensorWithTFDims (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes_test.cc | 1531 void AddTestTensorWithTFDims( in AddTestTensorWithTFDims() function in tensorflow::tensorrt::convert::OpConverterTest 1566 AddTestTensorWithTFDims(name, dims_with_batch, trt_dtype); in AddTestTensor() 1845 AddTestTensorWithTFDims(name, partial_shape, trt_type, add_input_status); 4451 AddTestTensorWithTFDims("input", {-1, -1, -1, -1}, trt_type); in TEST_P()
|