pytorch
11614214 - [quant] Enable jit tracing on quantizable LSTM (#64438)

Commit
4 years ago
[quant] Enable jit tracing on quantizable LSTM (#64438) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64438 The quantizable LSTM didn't support jit tracing because it had several non taceable paths. We sacrifice some of the user experience to enable the tracing. The main UX feature removed is a user-friendly message when trying to access the backwards path in a bidirectional LSTM: When the bidirectional flag is `False`, we used to throw a nice error message when the user tried accessing backwards weights. Now the message is default (removed properties). Test Plan: `buck test mode/dev //caffe2/test:quantization -- test_custom_module_lstm` Reviewed By: mtl67 Differential Revision: D30732630 fbshipit-source-id: 443e351ebb0e2b636c86dea9691b9bf42ffe618f
Author
Zafar Takhirov
Parents
Loading