Searched refs:trt_dtype (Results 1 – 3 of 3) sorted by relevance
97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType() argument100 *trt_dtype = nvinfer1::DataType::kFLOAT; in ConvertDType()104 *trt_dtype = nvinfer1::DataType::kINT8; in ConvertDType()107 *trt_dtype = nvinfer1::DataType::kHALF; in ConvertDType()110 *trt_dtype = nvinfer1::DataType::kINT32; in ConvertDType()176 nvinfer1::DataType trt_dtype(nvinfer1::DataType::kFLOAT); in get() local177 TF_CHECK_OK(ConvertDType(this->at(key)->type(), &trt_dtype)); in get()178 return trt_dtype; in get()264 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties() argument267 TF_RETURN_IF_ERROR(ConvertDType(dtype, trt_dtype)); in ValidateTensorProperties()[all …]
167 string DebugString(const nvinfer1::DataType trt_dtype);269 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,432 nvinfer1::DataType trt_dtype; member
99 DataType TrtDataTypeToTf(nvinfer1::DataType trt_dtype) { in TrtDataTypeToTf() argument100 switch (trt_dtype) { in TrtDataTypeToTf()108 QCHECK(false) << "Unexpected data type " << static_cast<int>(trt_dtype); in TrtDataTypeToTf()1106 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local1108 const DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in CheckDataTypeMatches()1196 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor() argument1197 DataType tf_dtype = TrtDataTypeToTf(trt_dtype); in AddTestTensor()1208 converter_->AddInputTensor(name, trt_dtype, trt_dims, batch_size)); in AddTestTensor()