morpheus.utils.onnx_to_trt

Functions

gen_engine(config) This class converts an Onnx model to a TRT model.
gen_engine(config)[source]

This class converts an Onnx model to a TRT model.

Parameters
config : morpheus.config.ConfigOnnxToTRT

ONNX to TRT generator configuration.

Previous morpheus.utils.nvt.transforms
Next morpheus.utils.producer_consumer_queue
© Copyright 2024, NVIDIA. Last updated on Apr 11, 2024.