Home
last modified time | relevance | path

Searched refs:AddTestTensorWithTFDims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1531 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()