%VjijdddlmZddlZdgZGddejjjjZdS))AnyNLSTMceZdZdZejjjjZ de fdZ e de de ddfdZe dedd ejjjjddfd ZdS) rarA quantized long short-term memory (LSTM). For the description and the argument types, please, refer to :class:`~torch.nn.LSTM` Attributes: layers : instances of the `_LSTMLayer` .. note:: To access the weights and biases, you need to access them per layer. See examples in :class:`~torch.ao.nn.quantizable.LSTM` Examples:: >>> # xdoctest: +SKIP >>> custom_module_config = { ... 'float_to_observed_custom_module_class': { ... nn.LSTM: nn.quantizable.LSTM, ... }, ... 'observed_to_quantized_custom_module_class': { ... nn.quantizable.LSTM: nn.quantized.LSTM, ... } ... } >>> tq.prepare(model, prepare_custom_module_class=custom_module_config) >>> tq.convert(model, convert_custom_module_class=custom_module_config) returncdS)N QuantizedLSTM)selfs [/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/ao/nn/quantized/modules/rnn.py _get_namezLSTM._get_name'sargskwargsNc td)NzrIt looks like you are trying to convert a non-observed LSTM module. Please, see the examples on quantizable LSTMs.)NotImplementedError)clsrrs r from_floatzLSTM.from_float*s" 1   r rotherct||jsJtjj|dd}||_|S)NFT)inplaceremove_qconfig) isinstance _FLOAT_MODULEtorchao quantizationconvert __class__)rr converteds r from_observedzLSTM.from_observed4sQ%!233333H)11 52  " r )__name__ __module__ __qualname____doc__rrnn quantizablerrstrr classmethodrrtyper r r r rr s2HK+0M3 s c d   [ 4< 0G0LQW[r )typingrr__all__rr%r&rr r r r,sj    0000058; " '00000r