Package ai.djl.tensorrt.engine

package ai.djl.tensorrt.engine

Contains classes to interface with the underlying TensorRT Engine.

Classes

Class	Description
TrtEngine	The TrtEngine is an implementation of the Engine based on the TensorRT.
TrtEngineProvider	TrtEngineProvider is the TensorRT implementation of EngineProvider.
TrtModel	TrtModel is the TensorRT implementation of Model.
TrtNDArray	TrtNDArray is the TensorRT implementation of NDArray.
TrtNDManager	TrtNDManager is the TensorRT implementation of NDManager.
TrtSession	TrtSession represents the TensorRT's execution context.
TrtSymbolBlock	TrtSymbolBlock is the TensorRT implementation of SymbolBlock.