From 1469f610e1691253e17f9d70912eb3e659915fe7 Mon Sep 17 00:00:00 2001 From: "Zhang, Wuxun" Date: Thu, 29 Jun 2023 14:24:56 +0800 Subject: [PATCH] add Pow op in onednn graph --- .../elements/oneDNN/source/graph/ops/Pow.rst | 63 +++++++++++++++++++ .../oneDNN/source/graph/ops/index.rst | 1 + 2 files changed, 64 insertions(+) create mode 100644 source/elements/oneDNN/source/graph/ops/Pow.rst diff --git a/source/elements/oneDNN/source/graph/ops/Pow.rst b/source/elements/oneDNN/source/graph/ops/Pow.rst new file mode 100644 index 000000000..f61ac2845 --- /dev/null +++ b/source/elements/oneDNN/source/graph/ops/Pow.rst @@ -0,0 +1,63 @@ +.. SPDX-FileCopyrightText: 2023 Intel Corporation +.. +.. SPDX-License-Identifier: CC-BY-4.0 + +.. include:: ../../replacements.inc.rst + +Pow +### + +## General + +Pow operation performs an element-wise power operation on a given input tensor +with a single value attribute beta as its exponent. It is based on the following +mathematical formula: + +.. math:: + + \dst_i = \src_i ^ \beta + +## Operation attributes + ++--------------+----------------+------------+--------------+-------------+ +| Attribute | Description | Value Type | Supported | Required or | +| Name | | | Values | Optional | ++==============+================+============+==============+=============+ +| |attr_beta| | exponent, | f32 | Arbitrary | Required | +| | :math:`\beta` | | f32 value | | +| | in the | | | | +| | formula. | | | | ++--------------+----------------+------------+--------------+-------------+ + +Inputs +====== + + +===== ============= ==================== +Index Argument Name Required or Optional +===== ============= ==================== +0 ``src`` Required +===== ============= ==================== + +Outputs +======= + + +===== ============= ==================== +Index Argument Name Required or Optional +===== ============= ==================== +0 ``dst`` Required +===== ============= ==================== + +Supported Data Types +******************** + +Pow operation supports the following data type combinations. + +==== ==== +Src Dst +==== ==== +f32 f32 +bf16 bf16 +f16 f16 +==== ==== diff --git a/source/elements/oneDNN/source/graph/ops/index.rst b/source/elements/oneDNN/source/graph/ops/index.rst index 4f484e0c2..06d83dcfa 100644 --- a/source/elements/oneDNN/source/graph/ops/index.rst +++ b/source/elements/oneDNN/source/graph/ops/index.rst @@ -60,6 +60,7 @@ subset of the operation set. Mish.rst MishBackward.rst Multiply.rst + Pow.rst PReLU.rst PReLUBackward.rst Quantize.rst