pytorch/torch/quantization
Supriya Rao 284ff04792 [quant] Support set API for EmbeddingBag quantization (#43433)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/43433

Add support for torch.quint8 dtype

Test Plan: Imported from OSS

Reviewed By: radkris-git

Differential Revision: D23277002

fbshipit-source-id: 4204bc62f124b4fd481aaa6aa47b9437978c43ee
2020-08-24 14:33:35 -07:00
..
fx [quant][graphmode][fx] Add support for general value ops (#43439) 2020-08-22 08:52:28 -07:00
__init__.py [quant] Expose register activation post process hook function to user (#42342) 2020-08-03 12:28:42 -07:00
_correct_bias.py Bias Correction Implementation (#41845) 2020-08-20 21:40:33 -07:00
_equalize.py cross_layer_equalization (#41685) 2020-07-22 08:39:23 -07:00
_learnable_fake_quantize.py Extending Learnable Fake Quantize module to support gradient scaling and factory (partial) construction (#41969) 2020-07-29 10:22:26 -07:00
_numeric_suite.py Remove unused Logger in get_matching_activations (#41023) 2020-07-07 00:33:07 -07:00
_quantize_fx.py [quant][graphmode][fx][test] Add per op test for graph mode quant on fx (#43229) 2020-08-20 17:32:02 -07:00
default_mappings.py [quant][graphmode][fx] Add support for instance_norm (#43377) 2020-08-21 18:32:50 -07:00
fake_quantize.py [quant][doc] Print more info for fake quantize module (#43031) 2020-08-13 20:27:36 -07:00
fuse_modules.py [quant] Make OP_LIST_TO_FUSER_METHOD public (#43286) 2020-08-20 20:19:13 -07:00
observer.py [quant] Enable from_float for quantized Embedding_Bag (#43176) 2020-08-21 11:46:03 -07:00
qconfig.py [quant] Enable from_float for quantized Embedding_Bag (#43176) 2020-08-21 11:46:03 -07:00
quantize_jit.py [quant][graphmode] Enable inplace option for top level API (#40414) 2020-06-23 16:42:48 -07:00
quantize.py [quant] Support set API for EmbeddingBag quantization (#43433) 2020-08-24 14:33:35 -07:00
stubs.py Factored out the default mappings 2019-10-03 11:52:21 -07:00