FuseCustomConfig¶
- class torch.ao.quantization.fx.custom_config.FuseCustomConfig[source][source]¶
fuse_fx()
的自定义配置。示例用法
fuse_custom_config = FuseCustomConfig().set_preserved_attributes(["attr1", "attr2"])
- classmethod from_dict(fuse_custom_config_dict)[source][source]¶
从包含以下项的字典创建
ConvertCustomConfig
“preserved_attributes”:即使不在
forward
中使用也会保留的属性列表此函数主要用于向后兼容,将来可能会被移除。
- 返回类型