
    `i:                     J    S SK r SSKJr  SSKJrJr  S SK Jr   " S S\5      rg)	    N   )QuantOperatorBase   )QuantizedValueQuantizedValueType)onnx_pbc                   4   ^  \ rS rSrU 4S jrU 4S jrSrU =r$ )QMaxPool   c                 $   > [         TU ]  X5        g )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      o/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxruntime_tools/quantization/operators/maxpool.pyr   QMaxPool.__init__   s    3    c                   > U R                   nUR                  S:X  d   eU R                  R                  S:  a  [        TU ]  5         g UR                  S   U R                  R                  ;  a!  U R                  =R                  U/-  sl        g U R                  R                  UR                  S      n[        UR                  S   UR                  S   S-   UR                  UR                  [        R                  5      nX0R                  R                  UR                  S   '   UR                  UR                  S'   UR                  UR                  S'   U R                  =R                  U/-  sl        g )NMaxPool   r   
_quantized)nodeop_type	quantizeropset_versionr   quantizeinputquantized_value_map	new_nodesr   output
scale_namezp_namer   Inputq_name)r   r   quantized_input_valuequantized_output_valuer   s       r   r   QMaxPool.quantize   s.   yy	)*)>>''",G
 ::a= B BBNN$$.$ !% B B4::a= Q!/AAQ]@]0E0P0PRgRoRo0B0H0H"J >T**4;;q>:-44

1/66A  TF* r    )__name__
__module____qualname____firstlineno__r   r   __static_attributes____classcell__)r   s   @r   r
   r
      s    4+ +r   r
   )	onnxbase_operatorr   quant_utilsr   r   r   
onnx_protor
   r*   r   r   <module>r5      s     , < &+  +r   