ISerializationConfig

tensorrt.SerializationFlag

可用于从引擎创建二进制文件的有效标志。

成员

EXCLUDE_WEIGHTS : 排除可以重新拟合的权重。

EXCLUDE_LEAN_RUNTIME : 从计划中排除精简运行时。

class tensorrt.ISerializationConfig

用于保存配置引擎以序列化二进制文件的属性的类。

__init__(*args, **kwargs)

clear_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) → bool

从配置中清除序列化标志。

get_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) → bool

检查是否设置了序列化标志。

set_flag(self: tensorrt.tensorrt.ISerializationConfig, flag: tensorrt.tensorrt.SerializationFlag) → bool

将输入的序列化标志添加到已启用的标志中。