pytorch/torch/testing/_internal
Supriya Rao e34c825b77 [quant][fx] Embedding quantization support (#46677)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46677

Add support for weight only embedding quantization

Test Plan:
python test/test_quantization.py TestQuantizeFxOps.test_qembedding_module

Imported from OSS

Reviewed By: jerryzh168

Differential Revision: D24463305

fbshipit-source-id: 2dba49d8a77cf237a8e6da2efdd83b1ebdc432d6
2020-10-22 17:59:52 -07:00
..
codegen Enable CUDA Fuser for ROCm (#45965) 2020-10-08 10:41:56 -07:00
data remediation of S205607 2020-07-17 17:19:47 -07:00
distributed Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
generated remediation of S205607 2020-07-17 17:19:47 -07:00
test_module remediation of S205607 2020-07-17 17:19:47 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
autocast_test_lists.py [PyTorch] Enable type check for autocast_test_lists (#45107) 2020-09-22 16:54:26 -07:00
common_cuda.py Adjust TF32 tests (#44240) 2020-09-24 10:25:58 -07:00
common_device_type.py Cusolver inverse check info (#46625) 2020-10-21 21:46:33 -07:00
common_distributed.py [RPC] print exception message on workers that run python functions (#46372) 2020-10-22 09:44:15 -07:00
common_methods_invocations.py [numpy] torch.sin : support and promote integer inputs to float (#45733) 2020-10-22 01:58:57 -07:00
common_nn.py Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
common_quantization.py [quant][fx] Embedding quantization support (#46677) 2020-10-22 17:59:52 -07:00
common_quantized.py minor style edits to torch/testing/_internal/common_quantized.py (#44807) 2020-09-17 08:02:43 -07:00
common_utils.py renaming env var IN_CIRCLECI to a broader name of IN_CI (#46567) 2020-10-20 08:25:39 -07:00
dist_utils.py [RPC profiling] Extend RPC profiling to support async function execution over RPC. (#44664) 2020-09-25 13:19:26 -07:00
expecttest.py Add some more docs to expecttest. (#46263) 2020-10-13 15:17:11 -07:00
hypothesis_utils.py [quant] conv_transpose1d_prepack / conv_transpose1d_unpack (#40360) 2020-09-11 04:12:28 -07:00
jit_metaprogramming_utils.py enable mypy check for jit_metaprogramming_utils (#44752) 2020-09-16 15:44:37 -07:00
jit_utils.py Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
te_utils.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00