o jD@sGdddZdS)c@s$eZdZddZddZddZdS)QuantOperatorBasecCs||_||_dS)N) quantizernode)selfonnx_quantizer onnx_noderc/home/kuhnn/.local/lib/python3.10/site-packages/onnxruntime/quantization/operators/base_operator.py__init__s zQuantOperatorBase.__init__cCs&|j|js dS|j|jjdS)NF)rshould_quantize_noderis_float_tensorinput)rrrrshould_quantizesz!QuantOperatorBase.should_quantizecCsLt|jjD]\}}|j|}|dur|jj|q|jj|jdS)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 r_dequantize_value new_nodesappend)r_ node_inputdequantize_noderrrquantize s  zQuantOperatorBase.quantizeN)__name__ __module__ __qualname__r rrrrrrrs rN)rrrrrs