U U?hDã@sGdd„dƒZdS)c@s$eZdZdd„Zdd„Zdd„ZdS)ÚQuantOperatorBasecCs||_||_dS)N)Ú quantizerÚnode)ÚselfZonnx_quantizerZ onnx_node©rúb/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/quantization/operators/base_operator.pyÚ__init__szQuantOperatorBase.__init__cCs&|j |j¡sdS|j |jjd¡S)NFé)rZshould_quantize_noderZis_float_tensorÚinput)rrrrÚshould_quantizesz!QuantOperatorBase.should_quantizecCsLt|jjƒD]*\}}|j |¡}|dk r |jj |¡q |jj |j¡dS)a‡ Given a node which does not support quantization, this method checks whether the input to this node is quantized and adds a DequantizeLinear node to dequantize this input back to FP32 parameter node: Current node parameter new_nodes_list: List of new nodes created before processing current node return: List of new nodes created N)Ú enumeraterr rZ_dequantize_valueZ new_nodesÚappend)rÚ_Z node_inputZdequantize_noderrrÚquantize s  zQuantOperatorBase.quantizeN)Ú__name__Ú __module__Ú __qualname__rr rrrrrrsrN)rrrrrÚó