
    /j                    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                  V    e Zd Zedd            Zedd            Zedd            ZdS )ConvreturnNonec                    t          j        g dg dg dg dg dggg                              t           j                  } t          j        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gg                              t           j                  }t          || |g|gd           t          j                            ddd	gd
gddgg d          }t          j        g dg dg dggg                              t           j                  }t          || |g|gd           d S )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         e/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/conv.pyexportzConv.export   sD   H 211111666666666

 
 &

 	
 H (''

 

 &

 	
 !K11:EQ 2 
 
  766666888:::999

 
 &

 	 	q6#$/		
 	
 	
 	
  ${44:EQ  5  
  
 H +**,,,---

 

 &

 	 	 q6&'2		
 	
 	
 	
 	
 	
    c            
        t          j        g dg dg dg dg dg dg dggg                              t           j                  } t          j        g dg dg dggg                              t           j                  }t          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                              t           j                  }t          || |g|gd           t          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                              t           j                  }t          || |g|gd           t          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                              t           j                  }t          || |g|gd(           d S ))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/   r1   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_paddingr3   )r   r   r;   r<   r=   r>   node_with_asymmetric_paddingy_with_asymmetric_paddings           r?   export_conv_with_strideszConv.export_conv_with_stridesa   s$   H 211111666666666666666

 
 &

 	
 H (''

 

 &

 	
 !K11:EQ 2 

 

  +**+++------		
 
 &

 	 	q6#$1		
 	
 	
 	
  ${44:EQ  5 
 
 
 
 H t

 

 &

 	 	 q6&'4		
 	
 	
 	
 (,{'<'<:EQ (= 
(
 
(
$ %'H tu		%
 %
 &

 	" 	(q6./@		
 	
 	
 	
 	
 	
rA   c            	        t          j        g dg dg dg dg dggg                              t           j                  } t          j        g dg dg dggg                              t           j                  }t          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                              t           j                  }t          || |g|gd           d S )Nr   r   r   r   r   r   r   r   r   r   
SAME_LOWERr   rC   )r   r   auto_padr   rD   rE   rF   )r#   r&   r-   test_conv_with_autopad_samer/   r3   )r   r   noder   s       r?   export_conv_with_autopad_samez"Conv.export_conv_with_autopad_same   s_   H 211111666666666

 
 &

 	
 H (''

 

 &

 	
 {$$:E!QF % 
 
 H!!!#6#6#68K8K8KLMN
 

&

 	
 	tQFQC6STTTTTTrA   N)r   r	   )__name__
__module____qualname__staticmethodr@   rL   rR    rA   r?   r   r      s        P
 P
 P
 \P
d v
 v
 v
 \v
p &U &U &U \&U &U &UrA   r   )

__future__r   numpyr4   r8   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rW   rA   r?   <module>r\      s    # " " " " "      , , , , , , . . . . . .tU tU tU tU tU4 tU tU tU tU tUrA   