Searched defs:trt_dtype (Results 1 – 4 of 4) sorted by relevance
102 string DebugString(const nvinfer1::DataType trt_dtype) { in DebugString()
221 nvinfer1::DataType trt_dtype(nvinfer1::DataType::kFLOAT); in get() local308 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties()766 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor()779 void setType(nvinfer1::DataType trt_dtype) override { in setType()838 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights()1087 TRT_ShapedWeights TrtWeightStore::GetTempWeights(nvinfer1::DataType trt_dtype, in GetTempWeights()1174 nvinfer1::DataType trt_dtype; in ConvertToTensorOrWeights() local4319 nvinfer1::DataType trt_dtype; in TfTensorToTrtWeights() local6366 nvinfer1::DataType trt_dtype; in ConvertGraphDefToEngine() local6409 nvinfer1::DataType trt_dtype; in ConvertGraphDefToEngine() local
453 nvinfer1::DataType trt_dtype; member
1453 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local1562 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor()