o toi73@s<ddlmZddlZddlmZddZGdddeZdS)) annotationsN)OpRunc XCs|durdd|jddD}|dur|jdd}|dur-dd|jddDd}|dur=dd|jddD}|jd|jd|ksR|jd|dkrtd|jd |jd |d |jd|jd|t|jdd|jd|fd |dkrg} d} |jd|} |jd} t|jdD]} t|D]|}|| | d|| |d| f}||| |d| }zt||d||d||| }Wn:ttfy }z+td|jd |jd |d |d|jd|jd|d|d|d|d|d |d}~ww| dkr| |jd7} | | |fqq|jdgt| ddjdd}| |d<tj t || ddj d}d}| D]'\} }||| | d|||jdf<||jd7}||jdkrnd}qH|durdd|jD}|jd|d<| t |} || 7}|S|ddkst |t|kr+t|}g}t|jd| }t|D]2\}}t|j||}||j||j|d|d|||||d|dqtj t ||j d}td|jdtd|jdg}t|D]\}}t|j||}|td|j||q||t |<|}|}|dvrg}g}tt|jdD]@}|j|}|||d||}|d|||||} |dkrj| dd}!n| d}!| |!}"||!||"q=||}t|jdkr|j\}#}$}%|\}&|\}'t|%|&|d|d|'d}(|d})d|&d}*|) }+|(|'},t |jd|jd|(f} |dur| ddddddf| d7<t|#D]}-t|jdD]}.t|$D]}/||.|.d|/|/df}0t|+|,|'D]}1|1|+|'}2|2|(krq |1|&d}td||*t ||*|&|%}3}4||-|-d|/|/d|3|4f}5|5j|0jkrt|* |dt |&|&|%||*|&}6}7|0dddd|6|7f}8|5j|8jkrtd|5jd|8jd|*d|d |&d!|%d"|'d t|5 d#|8 d$d%}9nt|5 d#|0 d$d%}9| |-|.|2f|97<q qqq| St|jd&kr|j\}#}$}%}:|\}&};|\}'}d|&dd|;d}*}?|) |> }+}@|(|'|=|<},}At |jd|jd|(|=f} |durG| d'| ddddddddf<t|#D]5}-t|jdD])}.t|$D] }/||.|.d|/|/df}0t|+|,|'D]}1|1|+|'}2|2|(krqs|1|&d}td||*t ||*|&|%}3}4t|@|A|<D]}B|B|@|<}C|C|=krq|B|;d}Dtd|D|?t |D|?|;|:}E}F||-|-d|/|/d|3|4|E|Ff}5|5j|0jkr^t|* |dt |&|&|%||*|&}6}7t|? |Ddt |;|;|:|D|?|;}G}H|0dddd|6|7|G|Hf}8|5j|8jkrOtd|5jd|8jd|*d(|?d|d)|Dd |&d*|;d!|%d+|:d"|'d,|<d t|5 d#|8 d$d%}9nt|5 d#|0 d$d%}9| |-|.|2|Cf|97<qqsq\qUqK| St|jd-kr|j\}#}$}%}:}I|\}&};}J|\}'}<}Kt|%|&|d|d|'d}(t|:|;|d|d&|<d}=t|I|J|d|d-|Kd}L|d|d|d})}>}Md|&dd|;dd|Jd}*}?}N|) |> |M }+}@}O|(|'|=|<|L|K},}A}Pt |jd|jd|(|=|Lf} |dur:| d.| ddddddddddf<t|#D]}-t|jdD]}.t|$D]}/||.|.d|/|/df}0t|+|,|'D]}1|1|+|'}2|2|(krvqf|1|&d}td||*t ||*|&|%}3}4t|@|A|<D]Z}B|B|@|<}C|C|=krq|B|;d}Dtd|D|?t |D|?|;|:}E}Ft|O|P|KD]+}Q|Q|O|K}R|R|LkrАq|Q|Jd}Std|S|Nt |S|N|J|I}T}U||-|-d|/|/d|3|4|E|F|T|Uf}5|5j|0jkrt|* |dt |&|&|%||*|&}6}7t|? |Ddt |;|;|:|D|?|;}G}Ht|N |Sdt |J|J|I|S|N|J}V}W|0dddd|6|7|G|H|V|Wf}8|5j|8jkrtd/gd|5jd|8jd|*d(|?d0|Nd|d)|Dd1|Sd |&d*|;d2|Jd!|%d+|:d3|Id"|'d,|<d4|Kd t|5 d#|8 d$d%}9nt|5 d#|0 d$d%}9| |-|.|2|C|Rf|97<qqqfqOqHq>| Std5|jd |jd|d6)7NcSg|]}dqS.0srrN/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_conv.py z(_conv_implementation..cSr)rrrrrr r r cSrrrrrrr r r rrzShape inconsistencies, X.shape=z , W.shape=z, group=z, W should be ./z , gx.shape=z , gw.shape=z , auto_pad=z , dilations=z, kernel_shape=z, pads=z , strides=)dtypecSrrrrrrr r Gr >VALID SAME_LOWER SAME_UPPERr)rrrzUnexpected shape z != z, oh=z, i=z, kh=z, sH=z, sth=)rr)rr)rr)rrrrz, ow=z, j=z, kw=z, sW=z, stw=)rrrrrz, oz=z, z=z, kz=z, sZ=z, stz=zThe convolution for X.shape=z is not implemented yet.)shape ValueErrornpprodrange_conv_implementation RuntimeErrorappendlistzerostuplerreshapeminmaxlen enumeratesliceintdotjoin)XXWBauto_pad dilationsgroup kernel_shapepadsstridesrestdmgdwbggxgwcve new_shapefinalpndnew_kernel_shapeiddinew_windicesheadtail target_size pad_neededpad_headpad_tailsNsCsHkhsthh_outh0ohbhehnnwcwiohrih1ih2imgjh1jh2w_r sWkwstww_outw0owbwewjowrjiw1iw2jw1jw2sZkzstzz_outz0ozbzezzozrziz1iz2jz1jz2rrr r s*6   $  & "  &&       $  (     "",$$ &   "   "( (  $$$(  ,   "   "   ".&@ 3rc@s$eZdZ       dddZdS)ConvNc CsBt|jdkrtd|jdt||||||||| |jfS)Nrz3X must have at least 3 dimensions but its shape is r)r(rrrastyper) selfr.r/r0r1r2r3r4r5r6rrr _run's  z Conv._run)NNNNNNN)__name__ __module__ __qualname__rrrrr r&sr) __future__rnumpyronnx.reference.op_runrrrrrrr s