o wiD @s@ddlmZddlZddlmZddlmZGdddeZdS)) annotationsN)OpRun)_conv_implementationc@s$eZdZ       dddZdS) QLinearConvNc Cs| p|j} | p |j} | p|j} | p|j} |p|j}|p|j}|tj}|dur,||8}|tj}|durbt |j dkr^|j d|j dkr^|j dfdt |j d}|| |8}n||8}t ||| | | | | || tj}t |dkrt|dkrtdt |dt |t |dkrtdt |dt |dt|d}||||}|dur||7}|jtjkrt|d d }nt|dd }t||jfS|jtjkrt|d d }nt|dd }t||jfS) Nr)rz4w_scale must be a scalar or a 1-D tensor. Got shape .z-w_scale elements must match output channels: z != )ri)auto_pad dilationsgroup kernel_shapepadsstridesastypenpint32lenshapereshapersizendim ValueError expand_dimsdtypeint8clipround)selfxx_scale x_zero_pointww_scale w_zero_pointy_scale y_zero_pointBr r rrrrXWmissingresRr/_/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/onnx/reference/ops/op_qlinear_conv.py_run sR        "   zQLinearConv._run)NNNNNNN)__name__ __module__ __qualname__r1r/r/r/r0r s r) __future__rnumpyronnx.reference.op_runronnx.reference.ops.op_convrrr/r/r/r0s