U qh`@s<ddlmZddlZddlmZddZGdddeZdS)) annotationsN)OpRunc 0Cs|dkr dd|jddD}|dkr6|jdd}| dkrZdd|jddDd} | dkrzdd|jddD} |dkrd}|dkrd}|jdd\} } |jd} |jdd}| |jd|ks| |dkrtd|jd |jd |d |jd| |}}| |dkr$td | |}|t|t||jdkrltd |jd|d|d || |f|t|f|}|dkrt| |t|f|}|| |f||}ddlm}t|jdkr^|jdd\}}|jdd\}}|\}}| \}}|\}}|d|d|d|d}}|t ||| d| d|dks|t ||| d| d|dkrt d| d | d }} tj | | ||f|j d}!|dk r|d|!ddddddddf<t td||td||\}"}#tj|#|"fdd}$t| D]"}%t| D]}&t| D]}'|'||&|krlqP|'|}(t|D]})|||)}*t|D]}+| ||+},t|$t}-|-dddddf|*||%|(ddddd|)|+f7<|-dddddf|,||%|(ddddd|)|+f7<|-dddddf|ddd|-dddddf<|-dddddf|ddd|-dddddf<t|-d}-t|-d}-|j||%|%d|'|'df|-dd}.t|.||&|'|ddddf}/t|/||%|(dddd|)|+f}/|!|%|&|)|+ft|/7<qq|qPqBq4|!St d|jd |jd|ddS)NcSsg|]}dqS.0srrE/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_deform_conv.py sz/_deform_conv_implementation..cSsg|]}dqS)rrrrrr r scSsg|]}dqSrrrrrr r srrzShape inconsistencies, X.shape=z , W.shape=z, group=.z;Number of input channels must be divisible by offset_group.z Offset shape z# is inconsistent with offset_group z and kernel shape ) GridSamplezKPadding, dilation, stride, and kernel shape incompatible with output shape.)dtype)rrr)Zaxis)Z align_cornerszThe convolution for X.shape=z, kernel_shape=z is not implemented yet.)shape ValueErrornpprodlenZreshapeZonesZonnx.reference.ops._op_listrint RuntimeErrorzerosrZmeshgridZarangestackrangecopyZastypefloatZ expand_dimsZflipevalmultiplysum)0XWoffsetBmask dilationsgroup kernel_shape offset_grouppadsstridesnZicocZ output_shapeZ ics_per_groupZ ocs_per_groupZics_per_offset_grouprZihiwZohZowZkhkwZsthZstwZdhZdwZkh_newZkw_newZbhZbwresZ kernel_pos_wZ kernel_pos_hZkernel_pos_wrt_first_elemZ batch_idxZoc_idxZic_idxZoffset_group_idxiZh_coordjZw_coordkernelZgrid_sample_outputZ conv_valuerrr _deform_conv_implementation s "  ". &   88 2r6c@seZdZdddZdS) DeformConvNc Cs>t|jdkr td|jdt||||||||| | | fS)Nrz3X must have at least 3 dimensions but its shape is r )rrrr6) selfr#r$r%r&r'r(r)r*r+r,r-rrr _runs$ zDeformConv._run)NNNNNNNN)__name__ __module__ __qualname__r9rrrr r7sr7) __future__rZnumpyrZonnx.reference.op_runrr6r7rrrr s