Searched defs:trt_dims (Results 1 – 2 of 2) sorted by relevance
204 nvinfer1::Dims trt_dims; in TensorShapeToTrtDims() local265 nvinfer1::Dims* trt_dims, int* batch_size) { in ValidateTensorProperties()645 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor()708 const nvinfer1::Dims& trt_dims, in TRT_TensorOrWeights()922 nvinfer1::Dims trt_dims; in ConvertToTensorOrWeights() local4109 nvinfer1::Dims trt_dims; in ConvertGraphDefToEngine() local
1205 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestTensor() local1218 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestWeights() local