
    bIg                    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                  B    e Zd Zedd       Zedd       Zedd       Zy)Convc                    t        j                  g dg dg dg dg dggg      j                  t         j                        } t        j                  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gg      j                  t         j                        }t        || |g|gd       t        j
                  j                  ddd	gd
gddgg d      }t        j                  g dg dg dggg      j                  t         j                        }t        || |g|gd       y )Ng              ?g       @g      @g      @g      @g      @g      @g       @g      "@g      $@g      &@      (@g      *@g      ,@g      .@g      0@g      1@g      2@g      3@g      4@      5@g      6@g      7@      8@r
   r
   r
   r   xWy      r   r   r   )inputsoutputskernel_shapepads)r   r         ;@     @@r   )r         K@     O@      R@g     I@)r         X@      [@     @]@     @T@)g     @W@      b@      c@     @d@     [@)r!   r)   r$        ^@      U@test_basic_conv_with_paddingr   r   namer   r   r   r   )r   r    r!   )r"   r#   r$   )r&   r'   r(   test_basic_conv_without_paddingnparrayastypefloat32onnxhelper	make_noder   )r   r   node_with_paddingy_with_paddingnode_without_paddingy_without_paddings         ]/var/www/html/answerous/venv/lib/python3.12/site-packages/onnx/backend/test/case/node/conv.pyexportzConv.export   s   HH 21666

 &
 	
 HH (''

 &
 	
 !KK11:EQ 2 
  768:9

 &
 	 	q6#$/		
  ${{44:EQ  5  
 HH +,-

 &
 	 	 q6&'2		
    c            
     
   t        j                  g dg dg dg dg dg dg dggg      j                  t         j                        } t        j                  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ddg      }t        j                  g dg dg dg dggg      j                  t         j                        }t        || |g|gd       t        j
                  j                  d	d
dgdgddgg dddg      }t        j                  ddgddgddgggg      j                  t         j                        }t        || |g|gd       t        j
                  j                  d	d
dgdgddgg dddg      }t        j                  d d!gd"d#gd$d%gd&d'gggg      j                  t         j                        }t        || |g|gd(       y ))Nr	   r   r   r   r   )g      9@g      :@r   g      <@g      =@)g      >@g      ?@g      @@r   g      A@r   r   r   r   r   r   r      )r   r   r   r   stridesr   r   r   r    r#   r%   )r*   g     h@g     a@)g      \@g      f@g      _@test_conv_with_strides_paddingr-   r/   r   r!   r&   r(   g     @m@g     o@!test_conv_with_strides_no_padding)r   r   r   r   r   r   r"   r$   g     g@g     i@g     `e@g     f@-test_conv_with_strides_and_asymmetric_paddingr1   )r   r   r9   r:   r;   r<   node_with_asymmetric_paddingy_with_asymmetric_paddings           r=   export_conv_with_strideszConv.export_conv_with_stridesa   sv   HH 2166666

 &
 	
 HH (''

 &
 	
 !KK11:EQ 2 

  ++--		
 &
 	 	q6#$1		
  ${{44:EQ  5 
 
 HH t

 &
 	 	 q6&'4		
 (,{{'<'<:EQ (= 
(
$ %'HH tu		%
 &
 	" 	(q6./@		
r?   c            	        t        j                  g dg dg dg dg dggg      j                  t         j                        } t        j                  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ddg      }t        j                  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   
SAME_LOWERr   rA   )r   r   auto_padr   rB   rC   rD   )r!   r$   r+   test_conv_with_autopad_samer-   r1   )r   r   noder   s       r=   export_conv_with_autopad_samez"Conv.export_conv_with_autopad_same   s   HH 21666

 &
 	
 HH (''

 &
 	
 {{$$:E!QF % 
 HH!#68KLMN

&
 	
 	tQFQC6STr?   N)returnNone)__name__
__module____qualname__staticmethodr>   rJ   rP    r?   r=   r   r      sD    P
 P
d v
 v
p &U &Ur?   r   )

__future__r   numpyr2   r6   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rW   r?   r=   <module>r\      s&    #   , .tU4 tUr?   