
    hp                     T    S SK JrJr  S SKJr  S SKr  " S S\5      r " S S5      rg)    )ABCabstractmethod)AnyNc            	           \ rS rSr\S\R                  R                  S\S\S\R                  R                  4S j5       r	Sr
g)		Quantizer   modelargskwargsreturnc                     g N selfr	   r
   r   s       V/home/james-whalen/.local/lib/python3.13/site-packages/torchao/quantization/unified.pyquantizeQuantizer.quantize        	    r   N)__name__
__module____qualname____firstlineno__r   torchnnModuler   r   __static_attributes__r   r   r   r   r      sD    XX__-0<?	 r   r   c            	           \ rS rSr\S\R                  R                  S\S\S\R                  R                  4S j5       r	\S\R                  R                  S\S\S\R                  R                  4S j5       r
Srg	)
TwoStepQuantizer   r	   r
   r   r   c                     g r   r   r   s       r   prepareTwoStepQuantizer.prepare   r   r   c                     g r   r   r   s       r   convertTwoStepQuantizer.convert%   r   r   r   N)r   r   r   r   r   r   r   r   r   r#   r&   r   r   r   r   r    r       s    XX__-0<?	 
 XX__-0<?	 r   r    )abcr   r   typingr   r   r   r    r   r   r   <module>r*      s+    $    r   