
    he                     B    S SK rS SKrS SKJr  S SKJr   " S S\5      rg)    N)Base)expectc                   P    \ rS rSr\SS j5       r\SS j5       r\SS j5       rSrg)
DeformConv   Nc                  F   [         R                  " S5      R                  [         R                  5      n SU l        [         R
                  " S[         R                  S9n[         R                  " S[         R                  S9nSUS'   SUS	'   [        R                  R                  S
/ SQS/SS// SQS9n[         R                  " / SQ/ SQ/ SQ/ SQ///5      R                  [         R                  5      n[        UXU/U/SS9  [         R                  " S[         R                  S9nSUS'   SUS'   [        R                  R                  S
/ SQS/SS// SQS9n[         R                  " SS/SS////5      R                  [         R                  5      n[        UXU/U/SS9  g )N	      r      r   r   r      r   dtype)r         r         ?r   r   r   r   皙)r      r   r   r   )XWoffset_with_paddingY_with_paddingr   )r   r   r   r   inputsoutputskernel_shapepads)g        g      ?      @g       @)r           @'@g      @)g      "@      4@      8@      *@)g      @r%   g      .@r!   #test_basic_deform_conv_with_paddingr   r   namer   r   r   r   r   r   r   r   )r   r   offset_without_paddingY_without_paddingg      #@r"   r#   r$   &test_basic_deform_conv_without_paddingnparangeastypefloat32shapeoneszerosonnxhelper	make_nodearrayr   )r   r   r   node_with_paddingr   r+   node_without_paddingr,   s           `/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/deformconv.pyexportDeformConv.export   s   IIaL

+GGL

3 !hh|2::F  		

  		
 !KK114%&Q 2 
  --/.		
 &
 	 	-.#$6		
 "$,bjj!I  		

  		
  ${{447()Q  5  
 HH dt	
 &
 	 	 01&'9		
    c                     [         R                  " S5      R                  [         R                  5      n SU l        [         R
                  " S[         R                  S9n[         R
                  " S[         R                  S9n[         R                  " S[         R                  S9nSUS'   S	US
'   [         R
                  " S[         R                  S9nSUS'   [        R                  R                  S/ SQS/SS// SQS9n[         R                  " SS/SS////5      R                  [         R                  5      n[        UXX2U/U/SS9  g )Nr	   r
   r   r   )r   r)   r   r   r   r*   )r   r   r   r   g?)r   r   r   r   r   )r   r   offsetBmaskYr   r   g      %@g)@g      5@gffffff3@test_deform_conv_with_mask_biasr'   r.   )r   r   rB   rA   rC   noderD   s          r<    export_deformconv_with_mask_bias+DeformConv.export_deformconv_with_mask_biasW   s4   IIaL

+GGL

3GGD

+,bjj9  		

  		
 ww|2::6Z{{$$4EQ % 
 HH tt	
 &
 	
 	&T*C2		
r?   c            	         [         R                  " S[         R                  S9n [         R                  " [         R                  " S5      R                  [         R                  5      S5      U S'   [         R                  " [         R                  " SSS5      R                  [         R                  5      S5      U S'   SU l        [         R                  " S	[         R                  S9n[         R                  " S
[         R                  S9nSUS'   SUS'   [        R                  R                  S/ SQS/SS// SQSS9n[         R                  " SS/SS////5      R                  [         R                  5      n[        UXU/U/SS9  g )N)r   r   r   r   r   r	   )r   r   )r   r   r   )r   r   )r   r   r   r   )r      r   r   r   r   r   )r      r   r   r   )r   r   rA   rD   r   )r   r   r   r   offset_groupg     @@g@@g      @@,test_deform_conv_with_multiple_offset_groupsr'   )r/   r5   r2   reshaper0   r1   r3   r4   r6   r7   r8   r9   r   )r   r   rA   rF   rD   s        r<   -export_deformconv_with_multiple_offset_groups8DeformConv.export_deformconv_with_multiple_offset_groups   sU   HH\4**RYYq\00<fE$**RYYq"b188DfM$GGL

3-rzz:  		
  		
 {{$$'EQ % 
 HH tt	
 &
 	
 	&>C?		
r?    )returnN)	__name__
__module____qualname____firstlineno__staticmethodr=   rG   rP   __static_attributes__rR   r?   r<   r   r      s@    G
 G
R '
 '
R (
 (
r?   r   )numpyr/   r6   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rR   r?   r<   <module>r]      s!   
   , .^
 ^
r?   