pytorch/torch/ao/quantization
Ben Koopman a58ff186e8 [quant][embedding qat] Add basic EmbeddingBag QAT fakeQuant workflow (#65443)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65443

Test Plan: Imported from OSS

Reviewed By: dagitses, supriyar

Differential Revision: D31456445

Pulled By: b-koopman

fbshipit-source-id: 0edda6e272d9005fce65f2ba6a5e6abc831836de
2021-10-07 20:19:29 -07:00
..
fx [quant][fx] Don't assume bias is a keyword argument (#61647) 2021-10-06 07:25:47 -07:00
__init__.py [quant] Add imports to the torch/ao/quantization/__init__.py (#64911) 2021-09-29 19:08:45 -07:00
_correct_bias.py [quant] Fix the parts that were missing after initial migration (#66058) 2021-10-05 11:45:37 -07:00
_equalize.py [quant] AO migration of the _correct_bias.py, _equalize.py, and _learnable_fake_quantize.py (#64917) 2021-09-15 18:15:39 -07:00
_learnable_fake_quantize.py [quant] Fix the parts that were missing after initial migration (#66058) 2021-10-05 11:45:37 -07:00
fake_quantize.py [quant][embedding qat] Add basic EmbeddingBag QAT fakeQuant workflow (#65443) 2021-10-07 20:19:29 -07:00
fuse_modules.py [quant] Fix the parts that were missing after initial migration (#66058) 2021-10-05 11:45:37 -07:00
fuser_method_mappings.py [quant] AO migration of the torch/quantization/utils.py (phase 1) (#64919) 2021-09-16 21:30:18 -07:00
observer.py [quant] Add support for quantization of Embedding{Bag} in dynamic quant APIs (#65674) 2021-10-06 23:19:38 -07:00
qconfig.py [quant][embedding qat] Add basic EmbeddingBag QAT fakeQuant workflow (#65443) 2021-10-07 20:19:29 -07:00
quant_type.py [quant] AO migration of the quant_types.py (phase 1) (#64916) 2021-09-15 17:30:00 -07:00
quantization_mappings.py [quant][embedding qat] Add basic EmbeddingBag QAT fakeQuant workflow (#65443) 2021-10-07 20:19:29 -07:00
quantize_fx.py [quant] Fix the parts that were missing after initial migration (#66058) 2021-10-05 11:45:37 -07:00
quantize_jit.py [quant] Fix the parts that were missing after initial migration (#66058) 2021-10-05 11:45:37 -07:00
quantize.py [quant] Add support for quantization of Embedding{Bag} in dynamic quant APIs (#65674) 2021-10-06 23:19:38 -07:00
stubs.py torch.ao migration: stubs.py phase 1 (#64861) 2021-09-13 08:40:29 -07:00
utils.py [quant] AO migration of the torch/quantization/utils.py (phase 1) (#64919) 2021-09-16 21:30:18 -07:00