
    `i`                     N    S SK r SSKJr  SSKJrJrJr  S SK Jr   " S S\5      r	g)	    N   )QuantOperatorBase   )QuantizedValueQuantizedValueTypeattribute_to_kwarg)onnx_pbc                   .   ^  \ rS rSrU 4S jrS rSrU =r$ )QSplit   c                 $   > [         TU ]  X5        g )N)super__init__)selfonnx_quantizer	onnx_node	__class__s      m/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxruntime_tools/quantization/operators/split.pyr   QSplit.__init__   s    3    c                    U R                   nU R                  R                  US/5      u  p#pESnUR                  S:w  a  UR                  S-   n0 nUR                   H  nUR                  [        U5      5        M     / n	UR                   HS  n
U
S-   nU	R                  U5        [        XUS   US   [        R                  5      nXR                  R                  U
'   MU     [        UR                  5      S:  a  UR                  UR                  SS  5      n[         R"                  R$                  " UR&                  X)U40 UD6nUR                  U5        U R                  =R(                  U-  sl        g )Nr    _quant	quantizedr   )node	quantizerquantize_inputsname	attributeupdater   outputappendr   r   Inputquantized_value_mapleninputextendonnxhelper	make_nodeop_type	new_nodes)r   r   quantized_input_nameszero_point_namesscale_namesnodesquantized_node_namekwargsr   quantized_output_namesoutput_namequantized_output_nameq_outputquantized_nodes                 r   quantizeQSplit.quantize   sS   yyFJnnFdFdeilmknFoC 99?"&))h"6IMM,Y78 ( "$;;K$/+$=!"))*?@%k+VW.ZjklZm&8&>&>@H>FNN..{; ' tzz?Q$9$@$@AB$P!..t||=R/BNFLN 	^$  E) r    )__name__
__module____qualname____firstlineno__r   r8   __static_attributes____classcell__)r   s   @r   r   r      s    4* *r   r   )
r(   base_operatorr   quant_utilsr   r   r   r	   
onnx_protor   r:   r   r   <module>rD      s!     , P P &* *r   