ISerializationConfig¶
- tensorrt.SerializationFlag¶
Valid flags that can be use to creating binary file from engine.
Members:
EXCLUDE_WEIGHTS : Exclude weights that can be refitted.
EXCLUDE_LEAN_RUNTIME : Exclude lean runtime from the plan.
- class tensorrt.ISerializationConfig¶
Class to hold properties for configuring an engine to serialize the binary.
- __init__(*args, **kwargs)¶
- clear_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) bool ¶
Clears the serialization flag from the config.
- get_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) bool ¶
Check if a serialization flag is set.
- set_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) bool ¶
Add the input serialization flag to the already enabled flags.