Searched refs:trt_convert_params (Results 1 – 2 of 2) sorted by relevance
178 trt_convert_params: trt.TrtConversionParams = None):181 output_tensors, model_latency, trt_convert_params)392 trt_convert_params: trt.TrtConversionParams,395 self._trt_convert_params = trt_convert_params397 self._converter = self._create_converter(trt_convert_params)401 def _create_converter(self, trt_convert_params: trt.TrtConversionParams):419 def trt_convert_params(self) -> trt.TrtConversionParams: member in _TrtModelHandlerBase451 def _create_converter(self, trt_convert_params: trt.TrtConversionParams):459 max_workspace_size_bytes=trt_convert_params.max_workspace_size_bytes,460 precision_mode=trt_convert_params.precision_mode,[all …]
114 if result.trt_convert_params is not None:115 r += [True] + list(result.trt_convert_params)