4 Commits

Author	SHA1	Message	Date
Jerry Zhang	0cba3e3704	[quant][graphmode][fx] Add support for qat convbn{relu}1d (#47248) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47248 Test Plan: Imported from OSS Reviewed By: vkuzo Differential Revision: D24696524 fbshipit-source-id: 684db12be201307acbdc89a44192cf2270491dba	2020-11-03 22:43:33 -08:00
Jerry Zhang	bd90379df5	[quant][graphmode][fx] Add support for additional_fuse_method_mapping (#46345) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46345 Allow user to add more fusion mappings Test Plan: Imported from OSS Reviewed By: vkuzo Differential Revision: D24317439 fbshipit-source-id: 3b144bbc305e41efbdf3e9fb25dbbeaad9e86c6a	2020-10-22 15:15:31 -07:00
Jerry Zhang	f9446cb15a	[quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46337 We plan to pass around the mappings instead of using global registration api to keep the mappings local to the transformations user is performing Test Plan: Imported from OSS Reviewed By: vkuzo Differential Revision: D24317436 fbshipit-source-id: 81569b88f05eeeaa9595447e482a12827aeb961f	2020-10-20 15:53:47 -07:00
Jerry Zhang	0c58a017bd	[quant][eagermode][refactor] Add set/get method for quantization and fusion mappings (#43990) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/43990 Allow user to register custom quantization and fusion patterns Test Plan: Imported from OSS Reviewed By: z-a-f Differential Revision: D23485344 fbshipit-source-id: 4f0174ee6d8000d83de0f73cb370e9a1941d54aa	2020-09-10 21:29:39 -07:00