Searched refs:ConvertDType (Results 1 – 1 of 1) sorted by relevance
97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType() function177 TF_CHECK_OK(ConvertDType(this->at(key)->type(), &trt_dtype)); in get()267 TF_RETURN_IF_ERROR(ConvertDType(dtype, trt_dtype)); in ValidateTensorProperties()626 TF_CHECK_OK(ConvertDType(type_, &trt_type)); in GetTrtWeights()1584 TF_RETURN_IF_ERROR(ConvertDType(weights.type_, &trt_dtype)); in BinaryTensorOpWeight()3057 TF_RETURN_IF_ERROR(ConvertDType(converted_dtype, &trt_dtype)); in TfTensorToTrtWeights()4140 TF_RETURN_IF_ERROR(ConvertDType(tf_dtype, &trt_dtype)); in ConvertGraphDefToEngine()