Searched refs:trt_type (Results 1 – 1 of 1) sorted by relevance
625 nvinfer1::DataType trt_type(nvinfer1::DataType::kFLOAT); in GetTrtWeights() local626 TF_CHECK_OK(ConvertDType(type_, &trt_type)); in GetTrtWeights()627 return nvinfer1::Weights{trt_type, GetValues(), count()}; in GetTrtWeights()