Andrew Or f81b4ae55c [Quant] Make quantizable LSTM scriptable (#83304) ... Summary: Previously `torch.nn.quantizable.LSTM` was not scriptable due to (1) the use of asterisk to unpack arguments, and (2) some arguments being Optional, which was not understood by setitem. This commit resolves both of these issues, enabling LSTM quantized through custom modules to work with TorchScript. Test Plan: python test/test_quantization.py TestQuantizedOps.test_custom_module_lstm Reviewers: jerryzh168, z-a-f Subscribers: jerryzh168, z-a-f, supriyar Tasks: https://github.com/pytorch/pytorch/issues/83211 https://github.com/pytorch/pytorch/issues/75042 Pull Request resolved: https://github.com/pytorch/pytorch/pull/83304 Approved by: https://github.com/jerryzh168		2022-08-12 14:45:09 +00:00
..
experimental	Implement QAT for APoT (#83282)	2022-08-12 04:00:06 +00:00
__init__.py
test_backend_config.py	[Quant][fx] Implement BackendConfig (part 1) (#81469)	2022-07-24 00:34:48 +00:00
test_docs.py	Set default qengine to QNNPACK on ARM for quantization tests (#83097)	2022-08-11 20:01:53 +00:00
test_quantized_functional.py
test_quantized_module.py	Add prelu op and module for quantized CPU backend (#73491)	2022-07-20 07:48:15 +00:00
test_quantized_op.py	[Quant] Make quantizable LSTM scriptable (#83304)	2022-08-12 14:45:09 +00:00
test_quantized_tensor.py
test_utils.py
test_workflow_module.py
test_workflow_ops.py