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.