o q::j<ã@sRddlmZddlZddlZddlmZddlmZd d d „Z Gd d „d eƒZ dS)é)Ú annotationsN)ÚBase)ÚexpectÚxú np.ndarrayÚyÚreturncCst ||¡ |j¡S)N)ÚnpÚpowerÚastypeÚdtype)rr©r úf/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/backend/test/case/node/pow.pyÚpow src@s6eZdZed dd„ƒZed dd„ƒZed dd„ƒZd S) ÚPowrÚNonecCs¼tjjdddgdgd}t gd¢¡ tj¡}t gd¢¡ tj¡}t||ƒ}t|||g|gdd t  d ¡  d d d ¡ tj¡}tj   d d d ¡ tj¡}t||ƒ}t|||g|gdd dS)NrrrÚz©ÚinputsÚoutputs©ééé©éééÚtest_pow_example©rrÚnameé<rrrÚtest_pow) ÚonnxÚhelperÚ make_noder Úarrayr Úfloat32rrÚarangeÚreshapeÚrandomÚrandn©Únoderrrr r rÚexportsý  z Pow.exportcCsÐtjjdddgdgd}t gd¢¡ tj¡}t d¡ tj¡}t||ƒ}t|||g|gdd tjjdddgdgd}t gd¢gd ¢g¡ tj¡}t gd¢¡ tj¡}t||ƒ}t|||g|gd d dS) NrrrrrrrÚtest_pow_bcast_scalarrrÚtest_pow_bcast_array) r#r$r%r r&r r'rrr,r r rÚexport_pow_broadcast$s$ý ý zPow.export_pow_broadcastcCs|tjjdddgdgd}t gd¢¡ tj¡}t gd¢¡ tj¡}t||ƒ}t |||g|gdd t gd¢¡ tj¡}t gd¢¡ tj¡}t||ƒ}t |||g|gd d t gd¢¡ tj¡}t gd¢¡ tj ¡}t||ƒ}t |||g|gd d t gd¢¡ tj ¡}t gd¢¡ tj¡}t||ƒ}t |||g|gd d t gd¢¡ tj¡}t gd¢¡ tj ¡}t||ƒ}t |||g|gd d t gd¢¡ tj¡}t gd¢¡ tj ¡}t||ƒ}t |||g|gdd t gd¢¡ tj¡}t gd¢¡ tj¡}t||ƒ}t |||g|gdd t gd¢¡ tj ¡}t gd¢¡ tj ¡}t||ƒ}t |||g|gdd dS)NrrrrrrrÚ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'Úint64rrÚint32Úuint64Úuint32r,r r rÚ export_types<sJý        zPow.export_typesN)rr)Ú__name__Ú __module__Ú __qualname__Ú staticmethodr.r1r>r r r rrs  r)rrrrrr) Ú __future__rÚnumpyr r#Úonnx.backend.test.case.baserÚonnx.backend.test.case.noderrrr r r rÚs