Jerry Zhang 901ba2be86 [quant][pt2e] Add support for conv transpose + bn + {relu} weights fusion in PTQ (#122046) ... Summary: also added some utils in xnnpack_quantizer_utils.py * annotate_conv_tranpsose_bn_relu and annotate_conv_transpose_bn -> this is for QAT * annotate_conv_transpose_relu conv_transpose + bn weights fusion is performed automatically and can not be disabled currently we can add support to allow disable this fusion later if needed Test Plan: python test/test_quantization.py -k test_conv_transpose_bn_fusion Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/122046 Approved by: https://github.com/andrewor14		2024-03-19 21:00:57 +00:00
..
ao_migration	ao migration: remove package test as this behavior is tested by other things (#94422)	2023-02-13 16:33:40 +00:00
bc	[BE] Enable ruff's UP rules and autoformat test/ (#105434)	2023-07-19 20:36:06 +00:00
core	Change default torch_function behavior to be disabled when torch_dispatch is defined (take 2) (#120632)	2024-03-09 01:08:37 +00:00
eager	[BE]: Update flake8 to v6.1.0 and fix lints (#116591)	2024-01-03 06:04:44 +00:00
fx	[BE]: Apply RUF025 dict.fromkeys preview rule (#118637)	2024-01-30 20:46:54 +00:00
jit	[quant] Remove deprecated torch.jit.quantized APIs (#118406)	2024-01-27 18:32:45 +00:00
pt2e	[quant][pt2e] Add support for conv transpose + bn + {relu} weights fusion in PTQ (#122046)	2024-03-19 21:00:57 +00:00
serialized
__init__.py