
    /j&                    \    d dl mZ d dlZd dlZd dlmZ d dlmZ d Z	 G d de          Z
dS )    )annotationsN)Base)expectc                \    t          j        | |                              | j                  S )N)nppowerastypedtype)xys     d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/onnx/backend/test/case/node/pow.pypowr      s"    8Aq>>  )))    c                  V    e Zd Zedd            Zedd            Zedd            ZdS )PowreturnNonec                    t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j	        d
          
                    ddd                              t          j                  }t          j                            ddd                              t          j                  }t          ||          }t          | ||g|gd	           d S )Nr   r   r   zinputsoutputs                  test_pow_exampler   r   name<   r   r   r   test_pow)onnxhelper	make_noder   arrayr	   float32r   r   arangereshaperandomrandnnoder   r   r   s       r   exportz
Pow.export   s&   {$$:E % 
 
 HYYY&&rz22HYYY&&rz221IItQFQC6HIIIIIbMM!!!Q**11"*==IOOAq!$$++BJ771IItQFQCjAAAAAAr   c                    t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        d                              t          j                  }t          ||          }t          | ||g|gd	           t           j                            dddgdg          } t          j        g dg d
g                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           d S )Nr   r   r   r   r   r   r   test_pow_bcast_scalarr"   r   test_pow_bcast_array)	r&   r'   r(   r   r)   r	   r*   r   r   r/   s       r   export_pow_broadcastzPow.export_pow_broadcast$   sG   {$$:E % 
 
 HYYY&&rz22HQKKrz**1IItQFQC6MNNNN{$$:E % 
 

 Hiii+,,33BJ??HYYY&&rz221IItQFQC6LMMMMMMr   c                 	   t           j                            dddgdg          } t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd
	           t          j        g d                              t          j                  }t          j        g d                              t          j
                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j
                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j                  }t          j        g d                              t          j                  }t          ||          }t          | ||g|gd	           t          j        g d                              t          j
                  }t          j        g d                              t          j
                  }t          ||          }t          | ||g|gd	           d S )Nr   r   r   r   r   r   r   test_pow_types_float32_int64r"   test_pow_types_int64_float32test_pow_types_float32_int32test_pow_types_int32_float32test_pow_types_float32_uint64test_pow_types_float32_uint32test_pow_types_int64_int64test_pow_types_int32_int32)r&   r'   r(   r   r)   r	   r*   int64r   r   int32uint64uint32r/   s       r   export_typeszPow.export_types<   s   {$$:E % 
 
 HYYY&&rz22HYYY&&rx001IItQFQC6TUUUUHYYY&&rx00HYYY&&rz221IItQFQC6TUUUUHYYY&&rz22HYYY&&rx001IItQFQC6TUUUUHYYY&&rx00HYYY&&rz221IItQFQC6TUUUUHYYY&&rz22HYYY&&ry111IItQFQC6UVVVVHYYY&&rz22HYYY&&ry111IItQFQC6UVVVVHYYY&&rx00HYYY&&rx001IItQFQC6RSSSSHYYY&&rx00HYYY&&rx001IItQFQC6RSSSSSSr   N)r   r   )__name__
__module____qualname__staticmethodr1   r5   rC    r   r   r   r      s        B B B \B" N N N \N. -T -T -T \-T -T -Tr   r   )
__future__r   numpyr   r&   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   rH   r   r   <module>rM      s    # " " " " "      , , , , , , . . . . . .* * *YT YT YT YT YT$ YT YT YT YT YTr   