18 lines
409 B
Python
18 lines
409 B
Python
# Copyright (c) ONNX Project Contributors
|
|
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
from __future__ import annotations
|
|
|
|
from warnings import catch_warnings, simplefilter
|
|
|
|
import numpy as np
|
|
|
|
from onnx.reference.op_run import OpRun
|
|
|
|
|
|
class Pow(OpRun):
|
|
def _run(self, a, b): # type: ignore
|
|
with catch_warnings():
|
|
simplefilter("ignore")
|
|
return (np.power(a, b).astype(a.dtype),)
|