
    /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                      e Zd Zed	d            Zed	d            Zed	d            Zed	d            Zed	d            ZdS )
ReduceL1returnNonec                 ,   g d} t          j        dgt           j                  }d}t          j                            dddgdg|	          }t          j        t          j        d
t          j        |           d
z   t           j	                  |           }t          j
        t          j        |          t          |          |d
k              }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j	                  }t          j
        t          j        |          t          |          |d
k              }t          |||g|gd           d S )N      r   r   dtyper   r   dataaxesreducedinputsoutputskeepdims   aaxisr   &test_reduce_l1_do_not_keepdims_exampler   r   name
   %test_reduce_l1_do_not_keepdims_randomnparrayint64onnxhelper	make_nodereshapearangeprodfloat32sumabstupler   randomseeduniformastypeshaper   r   noder   r   s         i/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/reducel1.pyexport_do_not_keepdimszReduceL1.export_do_not_keepdims   s{   		x28,,,{$$F#K	 % 
 
 z")Arwu~~'9LLLeTT &26$<<eDkkHPQMRRR 	$<I9		
 	
 	
 	
 		qy  b%0077
CC&26$<<eDkkHPQMRRR$<I8		
 	
 	
 	
 	
 	
    c                 ,   g d} t          j        dgt           j                  }d}t          j                            dddgdg|	          }t          j        t          j        dt          j        |           dz   t           j	                  |           }t          j
        t          j        |          t          |          |dk    
          }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j	                  }t          j
        t          j        |          t          |          |dk    
          }t          |||g|gd           d S )Nr   r   r   r   r   r   r   r   r   r    test_reduce_l1_keep_dims_exampler   r   r   r   test_reduce_l1_keep_dims_randomr!   r3   s         r6   export_keepdimszReduceL1.export_keepdims5   s{   		x28,,,{$$F#K	 % 
 
 z")Arwu~~'9LLLeTT &26$<<eDkkHPQMRRR 	$<I3		
 	
 	
 	
 		qy  b%0077
CC&26$<<eDkkHPQMRRR$<I2		
 	
 	
 	
 	
 	
r8   c                    g d} t          j        g t           j                  }d}t          j                            dddgdg|          }t          j        t          j        dt          j        |           dz   t           j	                  |           }t          j
        t          j        |          d |dk    	          }t          |||g|gd
           t           j                            d           t           j                            dd|                               t           j	                  }t          j
        t          j        |          d |dk    	          }t          |||g|gd           d S )Nr   r   r   r   r   r   r   r   r   ,test_reduce_l1_default_axes_keepdims_exampler   r   r   r   +test_reduce_l1_default_axes_keepdims_random)r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r   r/   r0   r1   r2   r3   s         r6   export_default_axes_keepdimsz%ReduceL1.export_default_axes_keepdims\   sh   		x"(+++{$$/)x % 
 
 z")Arwu~~'9LLLeTT &26$<<dX]KKK 	$<I?		
 	
 	
 	
 		qy  b%0077
CC&26$<<dX]KKK$<I>		
 	
 	
 	
 	
 	
r8   c                 ,   g d} t          j        dgt           j                  }d}t          j                            dddgdg|	          }t          j        t          j        dt          j        |           dz   t           j	                  |           }t          j
        t          j        |          t          |          |dk    
          }t          |||g|gd           t           j                            d           t           j                            dd|                               t           j	                  }t          j
        t          j        |          t          |          |dk    
          }t          |||g|gd           d S )Nr   r   r   r   r   r   r   r   r   .test_reduce_l1_negative_axes_keep_dims_exampler   r   r   r   -test_reduce_l1_negative_axes_keep_dims_randomr!   r3   s         r6   export_negative_axes_keepdimsz&ReduceL1.export_negative_axes_keepdims   s{   		xBH---{$$F#K	 % 
 
 z")Arwu~~'9LLLeTT &26$<<eDkkHPQMRRR 	$<IA		
 	
 	
 	
 		qy  b%0077
CC&26$<<eDkkHPQMRRR$<I@		
 	
 	
 	
 	
 	
r8   c                    g d} d}g d}t           j                            dddgdg|          }t          j        g t          j        	                              |           }t          j        dgt          j        	          }t          j        t          j        |t          j        	                    }t          |||g|gd
           d S )N)r   r      r   )r   r   rG   r   r   r   r   r   r   test_reduce_l1_empty_setr   )
r%   r&   r'   r"   r#   r+   r(   r$   zerosr   )r4   r   reduced_shaper5   r   r   r   s          r6   export_empty_setzReduceL1.export_empty_set   s    		!		{$$F#K	 % 
 
 x"*---55e<<x28,,,(28MDDDEE$<I+		
 	
 	
 	
 	
 	
r8   N)r   r	   )	__name__
__module____qualname__staticmethodr7   r<   r@   rE   rK    r8   r6   r   r      s        $
 $
 $
 \$
L $
 $
 $
 \$
L !
 !
 !
 \!
F $
 $
 $
 \$
L 
 
 
 \
 
 
r8   r   )

__future__r   numpyr"   r%   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   rP   r8   r6   <module>rU      s    # " " " " "      , , , , , , . . . . . .p
 p
 p
 p
 p
t p
 p
 p
 p
 p
r8   