pytorch/torch/nn/quantized/modules
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
..
__init__.py Document factory_kwargs in nn.Quantize + remove Attributes section (#59025) 2021-05-26 14:40:48 -07:00
activation.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
batchnorm.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
conv.py [PyTorch] Fix quantized Conv1d module parameters (#62356) 2021-07-30 12:27:52 -07:00
embedding_ops.py [quant][embedding qat] Add basic EmbeddingBag QAT fakeQuant workflow (#65443) 2021-10-07 20:19:29 -07:00
functional_modules.py
linear.py fix copy.deepcopy on LinearPackedParams (#64367) 2021-09-02 06:30:42 -07:00
normalization.py Support factory kwargs in torch.nn modules (#54508) 2021-04-22 16:16:53 -07:00
utils.py