Skip to content

Commit

Permalink
add Pow op in onednn graph
Browse files Browse the repository at this point in the history
  • Loading branch information
wuxun-zhang committed Jun 29, 2023
1 parent 2bf056f commit 1469f61
Show file tree
Hide file tree
Showing 2 changed files with 64 additions and 0 deletions.
63 changes: 63 additions & 0 deletions source/elements/oneDNN/source/graph/ops/Pow.rst
Original file line number Diff line number Diff line change
@@ -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
==== ====
1 change: 1 addition & 0 deletions source/elements/oneDNN/source/graph/ops/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@ subset of the operation set.
Mish.rst
MishBackward.rst
Multiply.rst
Pow.rst
PReLU.rst
PReLUBackward.rst
Quantize.rst
Expand Down

0 comments on commit 1469f61

Please sign in to comment.