Engine¶
- class
dragon.vm.tensorrt.
Engine
(
cuda_engine,
device_id=0
)[source]¶ The executing engine with bindings.
__init__¶
Properties¶
cuda_engine¶
Engine.
cuda_engine
¶Return the built cuda engine.
- Returns:
- tensorrt.ICudaEngine – The cuda engine.
Methods¶
run¶
Engine.
run
(
inputs,
optimization_profile=None
)[source]¶Execute the engine and return the results.
- Parameters:
- inputs (Union[Sequence, Dict]) – The input numpy arrays.
- optimization_profile (int, optional) – The index of optimization profile to use.
- Returns:
Sequence[numpy.ndarray] – The output arrays.