
    bIg|S                    N    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	y)    )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y
)ConvTransposec            	        t        j                  g dg dg dggg      j                  t         j                        } t        j                  g dg dg dgg dg dg dggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd       y )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       f/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/convtranspose.pyexportzConvTranspose.export   s   HHABC

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

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

&
 	
 HH %oG$oG
 &
 	
 HH A@@@@@BBB@ A@@@@@BBB@
: &
; 	
> {{$$c3Z#AbRSW % 
 	tQFQC6WX{{$$c3Z#APQSTv % 
 	tQFQC6NO{{$$#JEFaQq6 % 	
 	tQFQC6WXr;   c                 D   t        j                  g dg dg dggg      j                  t         j                        } t        j                  g dg dg dgg dg dg dggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd       y )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-   r5   s       r9   export_convtranspose_padsz'ConvTranspose.export_convtranspose_pads  s    HHABC

&
 	
 HH %oG$oG
 &
 	
 {{$$c3Z#A\ % 
 HH ('''')) (''''))
. &
/ 	
2 	tQFQC6OPr;   c                    t        j                  g dg dg dggg      j                  t         j                        } t        j                  ddgddgggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd       y )N)r   r   r   )r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   )	dilations)r   r]   r&   rC   r   )ra   rN   g     P@r@   rB   )r%   rG         S@r   r   )r   r   g      V@rU   ra   )r   r   r$   rE   r[   test_convtranspose_dilationsr)   r-   r5   s       r9   export_convtranspose_dilationsz,ConvTranspose.export_convtranspose_dilationsI  s    HHABC

&
 	
 HHc
S#J/01299"**E{{$$c3Z#1a& % 
 HH 66644

 &
 	
 	tQFQC6TUr;   c            
     4   t        j                  g dg dg dggg      j                  t         j                        } t        j                  g dg dg dgg dg dg dggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd       y )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#   rA   r%   r&   )r   r   r&   r   r   r   test_convtranspose_autopad_samer)   r-   r5   s       r9   !export_convtranspose_autopad_samez/ConvTranspose.export_convtranspose_autopad_samed  s   HHABC

&
 	
 HH %oG$oG
 &
 	
 {{$$c3Z#PQSTv % 
 HH 7676:8 7676:8
* &
+ 	
. 	tQFQC6WXr;   c            	     8   t        j                  g dg dg dgg dg dg dggg      j                  t         j                        } t        j                  g dg dg dggg dg dg dggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd       y )Nr	   r   r   r   r@   rA   r   r&   rB   r   rC   rD   r   r   r   r   r   r   groupr   r   r   r"   r'   r   rF   rL   r   rA   r   rT        @Q@rV   rI   r         R@rc   rd   rR   r!   r]   re   ru   rM   r   rM   rV   r$   rD   test_convtranspose_group_2r)   r-   r5   s       r9   export_convtranspose_group_2z*ConvTranspose.export_convtranspose_group_2  s"   HH %oG&(:<NO
 &
 	
 HH %oG %oG		
 &
 	
 {{$$_sCj3%q$QHH 23554 66766
& &
' 	
* 	tQFQC6R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      j                  t         j                        } t        j                  g d
g d
g d
ggg d
g d
g d
ggg      j                  t         j                        }t        j
                  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      j                  t         j                        }t        || |g|gd !       y )"Nr	   r   r   r   r   r   )rE   rF   r#   )r   rG   rH   )r%   rI   rJ   r   r   r   r   r   r   r   r   r   r   r"   r'   r   r   r   r   r   )rE   rO   r^   rP   r#   )rP   g      T@rv   g      U@rS   )ra   r   rw   ry   r   )rU   g      W@rx   g      X@rW   )r%   rW   r   rY   rJ   "test_convtranspose_group_2_image_3r)   r-   r5   s       r9   $export_convtranspose_group_2_image_3z2ConvTranspose.export_convtranspose_group_2_image_3  s   HH %oG&(:<NO
 ();=OP&(:<NO
 %oG&(:<NO
 &
 	
  HH %oG %oG		
 &
 	
 {{$$_sCj3%q$QHH 23554 66766$ 77976 66766$ 23554 66766C13
f &
g 	
j 	!Q!3W	
r;   N)returnNone)__name__
__module____qualname__staticmethodr:   r>   r   r   r   r   r   r   r    r;   r9   r   r      s    %L %LN O O NO NO` AY AYF +Q +QZ V V4 )Y )YV +T +TZ U
 U
r;   r   )

__future__r   numpyr.   r2   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r;   r9   <module>r      s$    #   , .G
D G
r;   