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
..
ao_migration
bc
core	[Quant] Make quantizable LSTM scriptable (#83304)	2022-08-12 14:45:09 +00:00
dbr
eager	Make M1 tests green (#82213)	2022-08-05 16:12:08 +00:00
fx	Set default qengine to QNNPACK on ARM for quantization tests (#83097)	2022-08-11 20:01:53 +00:00
jit	Skip tests that don't call gradcheck in slow gradcheck CI (#82117)	2022-07-25 21:33:52 +00:00
serialized
__init__.py