
    Ng)@                     J    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 )    N)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
dS )ConvTransposereturnNc            	      "   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       e/var/www/html/ai-engine/env/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)   r1   s       r5   export_convtranspose_1dz%ConvTranspose.export_convtranspose_1d5   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Nr7   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@g      1@      2@      3@)r    r         6@g      7@r"   )      9@g      :@r         <@g      =@)      >@      ?@g      @@r!   g      A@)     A@r        B@g      C@g     C@)      D@g     D@      E@g     E@      F@)     F@g      G@g     G@      H@g     H@)      I@     I@      J@g     J@      K@)     K@      L@g     L@      M@     M@r   r   r   r   r   )r
   r   r   r   r   r   r   )r   r   r   r   r!   r"   r#   )r   r!   rQ        O@      R@rO   r   )rE   rV   g     X@g      [@     @]@     @T@rJ   )rC   rP   rY        U@     @W@g      P@r!   )r   rF   rM   rO   rQ   rH   rA   )r    rJ        P@rW   g     S@rQ   rD   )rN   g      Z@     @d@     e@     @g@g      `@r\   )     V@r_         r@      s@     @t@g     k@     \@)g      ^@g     n@g     w@     x@     y@     q@g      b@)r`         g@rg   g     `r@rb   g      j@g     Z@)rN   g     Y@g     c@r]         e@rd   rT   )      N@     ^@g     g@     h@g     i@g     a@rW   )     `@     @q@     pz@g     {@     |@g     s@g     c@)g      l@rp   g     @g     Ȇ@     @g     @     Pp@)g     p@g     (@g     (@g      @g     ؋@     ؂@rb   )g     `h@re   rs   g     h@g     @      {@g     `k@)     @Z@g     j@rc        t@g     `u@g     l@rX   )rj   g     ^@r_   g      h@rl   g     `@g      Q@)g     @`@     p@g      y@rf        z@ra   g     @b@)g     @j@rx   g     @@g     Є@g     `@g     |@g     @m@)g      n@g     `~@g     @rq   g     0@     P@rw   )g     @e@g     u@ry   g     @g     @g      w@r_   )r`   g     f@rn   rg   ra   g     @h@g     X@)rI   rY   rk   g     _@g      `@rZ   rK   )g     @U@g     e@rr   g     p@     q@rh   r[   )rm   rz   rf   ro   rt   g     0r@g     `b@)g     b@g     r@rp   g     @}@g     }@     t@r]   )ru   g     j@r{   g     pt@rv   g      l@g     @\@)rR   g     [@ri   g     `e@r^   rX   rU   test_convtranspose_3dr%   r)   r1   s       r5   export_convtranspose_3dz%ConvTranspose.export_convtranspose_3dE   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Nr7   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)   )r2   r   r4   r3   s       r5   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Xr7   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)   r1   s       r5   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Pr7   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   rS   r#   r?   r   )rV   rG   g     P@r<   r>   )r"   rB         S@r   r   )r   r   g      V@rL   rV   )r   r   r!   r@   rQ   test_convtranspose_dilationsr%   r)   r1   s       r5   export_convtranspose_dilationsz,ConvTranspose.export_convtranspose_dilationsH  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Ur7   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    r=   r"   r#   )r   r   r#   r   r   r   test_convtranspose_autopad_samer%   r)   r1   s       r5   !export_convtranspose_autopad_samez/ConvTranspose.export_convtranspose_autopad_samec  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Xr7   )r   N)__name__
__module____qualname__staticmethodr6   r:   r}   r   r   r   r    r7   r5   r   r      s       %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 )Y )Yr7   r   )numpyr*   r.   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r7   r5   <module>r      s   
      , , , , , , . . . . . .AY AY AY AY AYD AY AY AY AY AYr7   