Searched defs:trt_dtype (Results 1 – 6 of 6) sorted by relevance
102 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights()190 nvinfer1::DataType trt_dtype, const DimsAdapter& dims) { in GetTempWeights()
55 string DebugString(const nvinfer1::DataType trt_dtype) { in DebugString()
262 nvinfer1::DataType trt_dtype; member
223 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties()934 nvinfer1::DataType trt_dtype; in ConvertToTensorOrWeights() local3446 nvinfer1::DataType trt_dtype; in TfTensorToTrtWeights() local5919 nvinfer1::DataType trt_dtype; in ConvertGraphDefToEngine() local5972 nvinfer1::DataType trt_dtype; in ConvertGraphDefToEngine() local
1257 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local1385 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor()
39 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor()63 void setType(nvinfer1::DataType trt_dtype) { trt_dtype_ = trt_dtype; } in setType()126 explicit ITensorProxy(nvinfer1::DataType trt_dtype, in ITensorProxy()