Searched defs:trt_dims (Results 1 – 5 of 5) sorted by relevance
143 nvinfer1::Dims trt_dims; in TensorShapeToTrtDims() local
195 Status TrtDimsToTensorShape(const std::vector<int>& trt_dims, in TrtDimsToTensorShape()
309 nvinfer1::Dims* trt_dims, int* batch_size) { in ValidateTensorProperties()766 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor()839 const nvinfer1::Dims& trt_dims, in TRT_TensorOrWeights()1175 nvinfer1::Dims trt_dims; in ConvertToTensorOrWeights() local5698 const nvinfer1::Dims trt_dims = params->inputs.at(0).GetTrtDims(); in ConvertArgMinMax() local6367 nvinfer1::Dims trt_dims; in ConvertGraphDefToEngine() local
1544 const nvinfer1::Dims trt_dims = in AddTestTensorWithTFDims() local1585 const nvinfer1::Dims trt_dims = GetTestDims(dims); in AddTestWeights() local
164 nvinfer1::Dims trt_dims; in SetTrtEngineInputs() local