FuseCustomConfig#

class torch.ao.quantization.fx.custom_config.FuseCustomConfig[source]#

為 fuse_fx() 自定義配置。

使用示例

fuse_custom_config = FuseCustomConfig().set_preserved_attributes(
    ["attr1", "attr2"]
)

classmethod from_dict(fuse_custom_config_dict)[source]#

從包含以下專案的字典建立 ConvertCustomConfig

“preserved_attributes”: 即使屬性不在 forward 中使用，也會保留的屬性列表

此函式主要用於向後相容，將來可能會被刪除。

set_preserved_attributes(attributes)[source]#

設定將保留在圖模組中的屬性名稱，即使它們未在模型的 forward 方法中使用。

將此 FuseCustomConfig 轉換為字典，字典中的專案在 from_dict() 中進行了描述。

文件