o
    iy                     @   s   d dl Z G dd de jjZG dd de jjZG dd de jjZG dd	 d	eZG d
d deZG dd deZG dd de jjZ	G dd de	Z
G dd de	Zdd Ze jfddZe jfddZdS )    Nc                   @      e Zd Zdd ZdS )QuantizedLinearc                 C      t d)Nzjtorch.jit.QuantizedLinear is no longer supported. Please use torch.ao.nn.quantized.dynamic.Linear instead.RuntimeErrorselfother r
   P/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/jit/quantized.py__init__      zQuantizedLinear.__init__N__name__
__module____qualname__r   r
   r
   r
   r   r          r   c                          e Zd Z fddZ  ZS )QuantizedLinearFP16c                    s   t    td)Nzrtorch.jit.QuantizedLinearFP16 is no longer supported. Please use the torch.ao.nn.quantized.dynamic.Linear instead.superr   r   r   	__class__r
   r   r      s   
zQuantizedLinearFP16.__init__r   r   r   r   __classcell__r
   r
   r   r   r          r   c                   @   r   )QuantizedRNNCellBasec                 C   r   )Nzttorch.jit.QuantizedRNNCellBase is no longer supported. Please use the torch.ao.nn.quantized.dynamic.RNNCell instead.r   r   r
   r
   r   r      r   zQuantizedRNNCellBase.__init__Nr   r
   r
   r
   r   r      r   r   c                   @   r   )QuantizedRNNCellc                 C   r   )Nzptorch.jit.QuantizedRNNCell is no longer supported. Please use the torch.ao.nn.quantized.dynamic.RNNCell instead.r   r   r
   r
   r   r   !   r   zQuantizedRNNCell.__init__Nr   r
   r
   r
   r   r       r   r   c                       r   )QuantizedLSTMCellc                       t  | td)Nzrtorch.jit.QuantizedLSTMCell is no longer supported. Please use the torch.ao.nn.quantized.dynamic.LSTMCell instead.r   r   r   r
   r   r   )      zQuantizedLSTMCell.__init__r   r
   r
   r   r   r   (   r   r   c                       r   )QuantizedGRUCellc                    r   )Nzptorch.jit.QuantizedGRUCell is no longer supported. Please use the torch.ao.nn.quantized.dynamic.GRUCell instead.r   r   r   r
   r   r   2   r    zQuantizedGRUCell.__init__r   r
   r
   r   r   r!   1   r   r!   c                   @   s   e Zd ZejfddZdS )QuantizedRNNBasec                 C   r   )Nzhtorch.jit.QuantizedRNNBase is no longer supported. Please use the torch.ao.nn.quantized.dynamic instead.r   r   r	   dtyper
   r
   r   r   ;   r   zQuantizedRNNBase.__init__N)r   r   r   torchint8r   r
   r
   r
   r   r"   :   s    r"   c                   @   r   )QuantizedLSTMc                 C   r   )Nzjtorch.jit.QuantizedLSTM is no longer supported. Please use the torch.ao.nn.quantized.dynamic.LSTM instead.r   r#   r
   r
   r   r   C   r   zQuantizedLSTM.__init__Nr   r
   r
   r
   r   r'   B   r   r'   c                   @   r   )QuantizedGRUc                 O   r   )Nzhtorch.jit.QuantizedGRU is no longer supported. Please use the torch.ao.nn.quantized.dynamic.GRU instead.r   )r   argskwargsr
   r
   r   r   K   r   zQuantizedGRU.__init__Nr   r
   r
   r
   r   r(   J   r   r(   c                 C   r   )Nzyquantize_rnn_cell_modules function is no longer supported. Please use torch.ao.quantization.quantize_dynamic API instead.r   )moduler
   r
   r   quantize_rnn_cell_modulesR   r   r,   c                 C   r   )Nzwquantize_linear_modules function is no longer supported. Please use torch.ao.quantization.quantize_dynamic API instead.r   r+   r$   r
   r
   r   quantize_linear_modulesY   r   r.   c                 C   r   )Nztquantize_rnn_modules function is no longer supported. Please use torch.ao.quantization.quantize_dynamic API instead.r   r-   r
   r
   r   quantize_rnn_modules`   r   r/   )r%   jitScriptModuler   r   r   r   r   r!   r"   r'   r(   r,   r&   r.   r/   r
   r
   r
   r   <module>   s   	
		