
    sh                     x    S SK r S/r " S S\ R                  R                  R                  R
                  5      rg)    NLSTMc                       \ rS rSrSr\R                  R                  R                  R                  r
S r\S 5       r\S 5       rSrg)r   
   a&  A 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)
c                     g)NQuantizedLSTM )selfs    u/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/ao/nn/quantized/modules/rnn.py	_get_nameLSTM._get_name%   s        c                     [        S5      e)NzrIt looks like you are trying to convert a non-observed LSTM module. Please, see the examples on quantizable LSTMs.)NotImplementedError)clsargskwargss      r
   
from_floatLSTM.from_float(   s     "1
 	
r   c                     [        XR                  5      (       d   e[        R                  R                  R                  USSS9nXl        U$ )NFT)inplaceremove_qconfig)
isinstance_FLOAT_MODULEtorchaoquantizationconvert	__class__)r   other	converteds      r
   from_observedLSTM.from_observed2   sM    %!2!23333HH))115 2 
	 "r   r   N)__name__
__module____qualname____firstlineno____doc__r   r   nnquantizabler   r   r   classmethodr   r!   __static_attributes__r   r   r
   r   r   
   sO    0 HHKK++00M 
 
  r   )r   __all__r   r(   r)   r   r   r   r
   <module>r-      s4     
/588;;""'' /r   