U qhS4@s<ddlmZddlZddlmZddZGdddeZdS)) annotationsN)OpRunc X*Cs|dkr dd|jddD}|dkr6|jdd}|dkrZdd|jddDd}|dkrzdd|jddD}|jd|jd|ks|jd|dkrtd|jd |jd |d |jd|jd|t|jdd|jd|fd |dkr4g} d} |jd|} |jd} t|jdD]} t|D]}|| | d|| |d| f}||| |d| }zt||d||d||| }Wn~ttfk r }zZtd|jd |jd |d |d|jd|jd|d|d|d|d|d |W5d}~XYnX| dkr:| |jd7} | | |fqDq6|jdft| ddjdd}| |d<tj t || ddj d}d}| D]N\} }||| | d|||jdf<||jd7}||jdkrd}q|dk r0dd|jD}|jd|d<| t |} || 7}|S|ddksTt |t|krht|}g}t|jd| }t|D]d\}}t|j||}||j||j|d|d|||||d|dq|tj t ||j d}td|jdtd|jdg}t|D]4\}}t|j||}|td|j||q||t |<|}|}|dkrg}g}tt|jdD]}|j|}|||d||}|d|||||} |dkr| dd}!n| d}!| |!}"||!||"q||}t|jdkr|j\}#}$}%|\}&|\}'t|%|&|d|d|'d}(|d})d|&d}*|) }+|(|'},t |jd|jd|(f} |dk r| ddddddf| d7<t|#D]}-t|jdD]}.t|$D]}/||.|.d|/|/df}0t|+|,|'D]Z}1|1|+|'}2|2|(krJq*|1|&d}td||*t ||*|&|%}3}4||-|-d|/|/d|3|4f}5|5j|0jkrRt|* |dt |&|&|%||*|&}6}7|0dddd|6|7f}8|5j|8jkr4td|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&k r$|j\}#}$}%}:|\}&};|\}'}d|&dd|;d}*}?|) |> }+}@|(|'|=|<},}At |jd|jd|(|=f} |dk r| d'| ddddddddf<t|#D]l}-t|jdD]T}.t|$D]B}/||.|.d|/|/df}0t|+|,|'D]}1|1|+|'}2|2|(k r q|1|&d}td||*t ||*|&|%}3}4t|@|A|<D]}B|B|@|<}C|C|=k rz qZ|B|;d}Dtd|D|?t |D|?|;|:}E}F||-|-d|/|/d|3|4|E|Ff}5|5j|0jk rt|* |dt |&|&|%||*|&}6}7t|? |Ddt |;|;|:|D|?|;}G}H|0dddd|6|7|G|Hf}8|5j|8jk rtd|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< qZ qqҐqĐq| 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} |dk r| d.| ddddddddddf<t|#D]"}-t|jdD] }.t|$D]}/||.|.d|/|/df}0t|+|,|'D]}1|1|+|'}2|2|(k r q|1|&d}td||*t ||*|&|%}3}4t|@|A|<D]h}B|B|@|<}C|C|=k rb qB|B|;d}Dtd|D|?t |D|?|;|:}E}Ft|O|P|KD] }Q|Q|O|K}R|R|Lk r q|Q|Jd}Std|S|Nt |S|N|J|I}T}U||-|-d|/|/d|3|4|E|F|T|Uf}5|5j|0jkrpt|* |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|8jkrRtd|5jd|8jd|*d(|?d/|Nd|d)|Dd0|Sd |&d*|;d1|Jd!|%d+|:d2|Id"|'d,|<d3|Kd #t|5 d#|8 d$d%}9nt|5 d#|0 d$d%}9| |-|.|2|C|Rf|97< q qB q q q q| Std4|jd |jd|d5dS)6NcSsg|]}dqS.0srr>/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_conv.py sz(_conv_implementation..cSsg|]}dqS)rrrrrr r scSsg|]}dqSrrrrrr r srrzShape 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=)dtypecSsg|]}dqSrrrrrr r Gs> SAME_LOWER SAME_UPPERVALIDr)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 RuntimeErrorappendlistzerostuplerZreshapeminmaxlen enumeratesliceintdot)XXWBauto_pad dilationsgroup kernel_shapepadsstridesrestdZmgZdwbgZgxZgwZcveZ new_shapefinalpZndZnew_kernel_shapeidZdiZnew_windicesheadtailZ target_sizeZ pad_neededZpad_headZpad_tailZsNZsCZsHZkhZsthZh_outZh0ZohZbhehnnwcwiohrZih1Zih2imgZjh1Zjh2Zw_r ZsWkwZstwZw_outZw0ZowZbwewZjowrjZiw1Ziw2Zjw1Zjw2ZsZZkzZstzZz_outZz0ZozZbzZezZzoZzrzZiz1Ziz2Zjz1Zjz2rrr r s*T  $ N & "  &&       $  (   "",2&$$ &   "   "( P,  $$$(  ,   "   "   ".&n2rc@seZdZdddZdS)ConvNc CsBt|jdkr td|jdt||||||||| |jfS)Nrz3X must have at least 3 dimensions but its shape is r )r$rrrZastyper) selfr)r*r+r,r-r.r/r0r1rrr _run+s$  z Conv._run)NNNNNNN)__name__ __module__ __qualname__rMrrrr rK*srK) __future__rZnumpyrZonnx.reference.op_runrrrKrrrr s  !