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
中未使用,也会保留的属性列表此函数主要用于向后兼容性,将来可能会删除。
- 返回类型