pytorch

556bb691 - [AO]Fix observed LSTM layer setup individually observed LSTM (#101299)

[AO]Fix observed LSTM layer setup individually observed LSTM (#101299) Summary: We have found that `_get_lstm_with_individually_observed_parts()` is missing setup step which sets up the LSTM layer state initializing weights and biases of this layer. This diff fixes the observed numerical discrepancy seen by CTRL team in using the above API. Test Plan: N3358643 Differential Revision: D45821681 Pull Request resolved: https://github.com/pytorch/pytorch/pull/101299 Approved by: https://github.com/andrewor14