
    i>                     B    S SK r S SKrS SKJr  SSKJr   " S S\5      rg)    N)rnn   )EagerModelBasec                   V    \ rS rSrS rS\R                  R                  4S jrS r	Sr
g)	LSTMModel   c                     g )N )selfs    d/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/examples/models/lstm/model.py__init__LSTMModel.__init__   s        returnc                     [         R                  " S5        [        R                  " SSS5      n[         R                  " S5        U$ )NzLoading LSTM model
      r   zLoaded LSTM model)logginginfor   LSTM)r   lstms     r   get_eager_modelLSTMModel.get_eager_model   s3    )*xxB"()r   c                     [         R                  " SSS5      n[         R                  " SSS5      n[         R                  " SSS5      nXU44$ )N      r   r   r   )torchrandn)r   input_tensorh0c0s       r   get_example_inputsLSTMModel.get_example_inputs   sE    {{1a,[[Ar"[[Ar"2h''r   r
   N)__name__
__module____qualname____firstlineno__r   r   nnModuler   r"   __static_attributes__r
   r   r   r   r      s      (r   r   )r   r   torch.nn.quantizable.modulesr   
model_baser   r   r
   r   r   <module>r-      s      , '( (r   