
    /j|S                    V    d dl mZ d dlZd dlZd dlmZ d dlmZ  G d de          Z	dS )    )annotationsN)Base)expectc                      e Zd Zedd            Zedd            Zedd            Zedd            Zedd            Zedd            Z	edd	            Z
edd
            Zedd            ZdS )ConvTransposereturnNonec            	     "   t          j        g dg dg dggg                              t           j                  } t          j        g dg dg dgg dg dg dggg                              t           j                  }t          j                            dddgdg          }t          j        g d	g d
g dg dg dgg d	g d
g dg dg dggg                              t           j                  }t          || |g|gd           d S )N              ?       @      @      @      @      @      @       @r   r   r   r   XWYr   r   r   r   r   r   r         .@      (@r         "@      5@      B@      ;@r   r          4@     @@      8@      *@r   r(   r!   r   r   test_convtransposeinputsoutputsnamenparrayastypefloat32onnxhelper	make_noder   xr   nodeys       n/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/convtranspose.pyexportzConvTranspose.export   s~   HABC
 

&

 	
 H %__oooG$__oooG
 
 &

 	
 {$$_sCj3%HHH 211333555555444 211333555555444
 
& &

' 	
* 	tQFQC6JKKKKKK    c                    t          j        g dgg                              t           j                  } t          j        g dg dgg                              t           j                  }t          j                            dddgdg          }t          j        g dg dgg                              t           j                  }t          || |g|gd	           d S )
Nr   r   r   r   r   r   r   test_convtranspose_1dr+   r/   r7   s       r;   export_convtranspose_1dz%ConvTranspose.export_convtranspose_1d6   s    H'())00<<H89::AAJ
 
 {$$_sCj3%HHH''')B)B)BCD
 

&

 	
 	tQFQC6MNNNNNNr=   c                    t          j        g dg dg dg dgg dg dg dg dgg d	g d
g dg dgggg                              t           j                  } t          j        g dg dg dgg dg dg dgg dg dg dggg dg dg dgg dg dg dgg dg dg dgggg                              t           j                  }t          j                            dddgdg          }t          j        g dg dg dg dg dg dgg dg dg dg dg dg dgg dg dg d g d!g d"g d#gg d$g d%g d&g d'g d(g d)gg d*g d+g d,g d-g d.g d/ggg dg dg dg dg dg dgg dg dg dg dg dg dgg dg dg d g d!g d"g d#gg d$g d%g d&g d'g d(g d)gg d*g d+g d,g d-g d.g d/gggg                              t           j                  }t          || |g|gd01           d S )2N)r   r   r   r   r   )r   r   r   r   r    )      $@      &@r   r(         ,@)r         0@      1@      2@      3@)r%   r!         6@      7@r'   )      9@      :@r#         <@g      =@)      >@      ?@g      @@r&   g      A@)     A@r"        B@g      C@     C@)      D@g     D@      E@     E@      F@)     F@g      G@g     G@      H@     H@)      I@     I@      J@g     J@      K@)     K@      L@     L@      M@     M@r   r   r   r   r   )r   r   r   r   r    r   r   )r   r   r!   r#   r&   r'   r(   )r   r&   r]        O@      R@r[   r#   )rN   rc   g     X@g      [@     @]@     @T@rT   )rK   r\   rf        U@     @W@g      P@r&   )r   rO   rX   r[   r]   rQ   rH   )r%   rT        P@rd   g     S@r]   rM   )rZ   g      Z@     @d@     e@     @g@g      `@ri   )     V@rl         r@      s@     @t@g     k@     \@)g      ^@g     n@g     w@     x@     y@     q@g      b@)rm         g@rt   g     `r@ro   g      j@g     Z@)rZ   g     Y@g     c@rj         e@rq   ra   )      N@     ^@g     g@     h@g     i@     a@rd   )     `@     @q@     pz@g     {@     |@g     s@g     c@)g      l@r~   g     @g     Ȇ@     @g     @     Pp@)g     p@g     (@g     (@g      @g     ؋@     ؂@ro   )g     `h@rr   r   g     h@g     @      {@g     `k@)     @Z@g     j@rp        t@g     `u@g     l@re   )rw   g     ^@rl   g      h@ry   g     `@g      Q@)g     @`@     p@g      y@rs        z@rn   g     @b@)g     @j@r   g     @@g     Є@g     `@g     |@g     @m@)g      n@g     `~@g     @r   g     0@     P@r   )g     @e@g     u@r   g     @g     @g      w@rl   )rm   g     f@r|   rt   rn   g     @h@g     X@)rS   rf   rx   g     _@      `@rg   rV   )g     @U@g     e@r   g     p@     q@ru   rh   )r{   r   rs   r}   r   g     0r@g     `b@)g     b@g     r@r~   g     @}@g     }@     t@rj   )r   g     j@r   g     pt@r   g      l@g     @\@)r^   g     [@rv   g     `e@rk   re   rb   test_convtranspose_3dr+   r/   r7   s       r;   export_convtranspose_3dz%ConvTranspose.export_convtranspose_3dF   sw   H
 655555::::::	 ;:::::::::::	 ;:::::::::::	
 
2 &

3 	
6 H
 ,OO+OO+OO
 )///???K(///???K )///???K(///???K(///???K
 
& &

' 	
* {$$_sCj3%HHH
 @??EEEFFFHHHFFFFFF GFFKKKLLLMMMLLLKKK LKKMMMMMMMMMMMMMMM LKKMMMMMMMMMMMMKKK JIIKKKMMMMMMMMMKKKC)X @??EEEFFFHHHFFFFFF GFFKKKLLLMMMLLLKKK LKKMMMMMMMMMMMMMMM LKKMMMMMMMMMMMMKKK JIIKKKMMMMMMMMMKKKC)WUWY
 Y
r &

s 	
v 	tQFQC6MNNNNNNr=   c                 ~   t          j        g dg dg dggg                              t           j                  } t          j        g dg dg dgg dg dg dggg                              t           j                  }t          j        g dg dg dg dg dg dg dg dg dg dg
g dg dg dg dg dg dg dg dg dg dg
gg                              t           j                  }t          j                            d	d
dgdgddgddg          }t          || |g|gd           t          j                            d	d
dgdgddgddg          }t          || |g|gd           t          j                            d	d
dgdgdddgddgddgddg          }t          || |g|gd           d S )Nr   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   r   r   r   r   r   r   r   r   r   r         
      )stridesoutput_shapetest_convtranspose_output_shaper+      )r   output_paddingtest_convtranspose_padtest)r.   r   r   kernel_shaper   test_convtranspose_kernel_shaper/   )r8   r   r:   r9   s       r;   export_convtranspose_attributesz-ConvTranspose.export_convtranspose_attributes   s   HABC
 

&

 	
 H %__oooG$__oooG
 
 &

 	
 H A@@@@@@@@@@@@@@@@@BBBBBBBBB@@@ A@@@@@@@@@@@@@@@@@BBBBBBBBB@@@
 
: &

; 	
> {$$c3Z#AbRSW % 
 
 	tQFQC6WXXXX{$$c3Z#APQSTv % 
 
 	tQFQC6NOOOO{$$#JEFaQq6 % 	
 	
 	tQFQC6WXXXXXXr=   c                 H   t          j        g dg dg dggg                              t           j                  } t          j        g dg dg dgg dg dg dggg                              t           j                  }t          j                            dddgdgd	d
gg d          }t          j        g dg dg dg dg dg dg dgg dg dg dg dg dg dg dggg                              t           j                  }t          || |g|gd           d S )Nr   r   r   r   r   r   r   r   r   r   )r   r   r   r   )r   pads)r   r   r   )r   r   r    )r(   r   r   test_convtranspose_padsr+   r/   r7   s       r;   export_convtranspose_padsz'ConvTranspose.export_convtranspose_pads  s   HABC
 

&

 	
 H %__oooG$__oooG
 
 &

 	
 {$$c3Z#A\\\ % 
 
 H ('''')))))) (''''))))))
 
. &

/ 	
2 	tQFQC6OPPPPPPr=   c                    t          j        g dg dg dggg                              t           j                  } t          j        ddgddgggg                              t           j                  }t          j                            dd	d
gdgddg          }t          j        g dg dg dg dg dggg                              t           j                  }t          || |g|gd           d S )N)r   r   r   )r    r   r   )r   r   r   r   r   r   r    r   r   r   r   r   )	dilations)r!   r_   r(   rE   r   )rc   rP   g     P@rB   rD   )r'   rI         S@r   r!   )r    r   g      V@rW   rc   )r   r   r&   rG   r]   test_convtranspose_dilationsr+   r/   r7   s       r;   export_convtranspose_dilationsz,ConvTranspose.export_convtranspose_dilationsI  s/   HABC
 

&

 	
 Hc
S#J/012299"*EE{$$c3Z#1a& % 
 
 H 655666666444444

 
 &

 	
 	tQFQC6TUUUUUUr=   c            
     8   t          j        g dg dg dggg                              t           j                  } t          j        g dg dg dgg dg dg dggg                              t           j                  }t          j                            dddgdgd	d
d
g          }t          j        g dg dg dg dg dg dgg dg dg dg dg dg dggg                              t           j                  }t          || |g|gd           d S )Nr   r   r   r   r   r   r   r   
SAME_UPPERr   )auto_padr   )r   r   r   r   r   r   )r   r   r   r   r   r   )r   r   r   r   r    r   )r    r    r%   rC   r'   r(   )r   r   r(   r   r   r   test_convtranspose_autopad_samer+   r/   r7   s       r;   !export_convtranspose_autopad_samez/ConvTranspose.export_convtranspose_autopad_samed  s   HABC
 

&

 	
 H %__oooG$__oooG
 
 &

 	
 {$$c3Z#PQSTv % 
 
 H 766666777666:::888 766666777666:::888
 
* &

+ 	
. 	tQFQC6WXXXXXXr=   c            	     <   t          j        g dg dg dgg dg dg dggg                              t           j                  } t          j        g dg dg dggg dg dg dggg                              t           j                  }t          j                            dd	d
gdgd          }t          j        g dg dg dg dg dgg dg dg dg dg dggg                              t           j                  }t          || |g|gd           d S )Nr   r   r   r    rB   rC   r   r(   rD   r   rE   rF   r   r   r   r   r   r   groupr   r   r   r$   r)   r    rH   rN   r!   rC   r!   rV        @Q@rX   rK   r"        R@re   rf   rT   r#   r_   rg   rw   rO   r   rO   rX   r&   rF   test_convtranspose_group_2r+   r/   r7   s       r;   export_convtranspose_group_2z*ConvTranspose.export_convtranspose_group_2  s   H %__oooG&&&(:(:(:<N<N<NO
 
 &

 	
 H %__oooG %__oooG		
 	
 &

 	
 {$$_sCj3%q$QQH 211333555555444 655666777666666
 
& &

' 	
* 	tQFQC6RSSSSSSr=   c                    t          j        g dg dg dgg dg dg dggg dg dg d	gg dg dg dggg dg dg dgg dg dg dggg                              t           j                  } t          j        g d
g d
g d
ggg d
g d
g d
ggg                              t           j                  }t          j                            dddgdgd          }t          j        g dg dg dg dg dgg dg dg dg dg dggg dg dg dg dg dgg dg dg dg dg dggg dg dg dg dg dgg dg dg dg dg dggg                              t           j                  }t          || |g|gd !           d S )"Nr   r   r   r   r   r   )rG   rH   r%   )r!   rI   rJ   )r'   rK   rL   r   r   r   r   r   r   r   r   r   r   r$   r)   r   r   r   r   r   )rG   rQ   r`   rR   r%   )rR   g      T@rx   g      U@rU   )rc   r   ry   r{   r   )rW   g      W@rz   g      X@rY   )r'   rY   r   r[   rL   "test_convtranspose_group_2_image_3r+   r/   r7   s       r;   $export_convtranspose_group_2_image_3z2ConvTranspose.export_convtranspose_group_2_image_3  s   H %__oooG&&&(:(:(:<N<N<NO
 ('');););=O=O=OP&&&(:(:(:<N<N<NO
 %__oooG&&&(:(:(:<N<N<NO
 
 &

 	
  H %__oooG %__oooG		
 	
 &

 	
 {$$_sCj3%q$QQH 211333555555444 655666777666666$ 766777999777666 655666777666666$ 211333555555444 655666777666666C13
 3
f &

g 	
j 	!Q!3W	
 	
 	
 	
 	
 	
r=   N)r   r	   )__name__
__module____qualname__staticmethodr<   r@   r   r   r   r   r   r   r    r=   r;   r   r      sf       %L %L %L \%LN O O O \O NO NO NO \NO` AY AY AY \AYF +Q +Q +Q \+QZ V V V \V4 )Y )Y )Y \)YV +T +T +T \+TZ U
 U
 U
 \U
 U
 U
r=   r   )

__future__r   numpyr0   r4   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r=   r;   <module>r      s    # " " " " "      , , , , , , . . . . . .G
 G
 G
 G
 G
D G
 G
 G
 G
 G
r=   