Aidge Export TensorRT API#

Export function#

Plugin helper#

The TensorRT export allow you to define plugins which will be automatically used when loadig the ONNX file.

The export define an helper command to generate the template of a plugin which you can then fill up.

Usage example:

python -m aidge_export_tensorrt.generate_plugin -n "test" -f "myExport"

This will create the plugin test in the folder myExport

myExport
+--- plugins
|   +--- test
|   |   +--- test_plugin.hpp
|   |   +--- test_plugin.cu
*--- ...