[oneDNN Graph] align with oneDNN v3.2 public release (#486)

* fix wrong argument names * add new select op and boolean data type * add Pow op in onednn graph
uxlfoundation · Jun 29, 2023 · da2fcac · da2fcac
1 parent e0eabad
commit da2fcac
Show file tree

Hide file tree

Showing 32 changed files with 386 additions and 224 deletions.
diff --git a/source/elements/oneDNN/source/graph/data_model.rst b/source/elements/oneDNN/source/graph/data_model.rst
@@ -39,6 +39,9 @@ it's suggested to refer to the definition page of each operation.
 
     u8
         8-bit unsigned integer.
+
+    boolean
+        Boolean data type. Size is C++ implementation defined.
 
 
 oneDNN Graph supports both public layout and |opaque| layout. When the

diff --git a/source/elements/oneDNN/source/graph/ops/Add.rst b/source/elements/oneDNN/source/graph/ops/Add.rst
@@ -68,10 +68,10 @@ Supported Data Types
 
 Add operation supports the following data type combinations.
 
-========= ===========
-Source0/1 Destination
-========= ===========
-f32       f32
-bf16      bf16
-f16       f16
-========= ===========
+============= ====
+Src_0 / Src_1 Dst
+============= ====
+f32           f32
+bf16          bf16
+f16           f16
+============= ====
diff --git a/source/elements/oneDNN/source/graph/ops/BatchNormTrainingBackward.rst b/source/elements/oneDNN/source/graph/ops/BatchNormTrainingBackward.rst
@@ -78,10 +78,10 @@ Supported Data Types
 BatchNormTrainingBackward operation supports the following data type
 combinations.
 
-==================== ================================================
-Src / Diff_dst / Dst Mean / Variance / Gamma / Diff_gamma / Diff_beta
-==================== ================================================
-f32                  f32
-bf16                 f32, bf16
-f16                  f32
-==================== ================================================
+========================= ================================================
+Src / Diff_dst / Diff_src Mean / Variance / Gamma / Diff_gamma / Diff_beta
+========================= ================================================
+f32                       f32
+bf16                      f32, bf16
+f16                       f32
+========================= ================================================
diff --git a/source/elements/oneDNN/source/graph/ops/ClampBackward.rst b/source/elements/oneDNN/source/graph/ops/ClampBackward.rst
@@ -93,10 +93,10 @@ Supported Data Types
 
 ClampBackward operation supports the following data type combinations.
 
-==== ======== ========
-Src  Diff_dst Diff_src
-==== ======== ========
-f32  f32      f32
-f16  f16      f16
-bf16 bf16     bf16
-==== ======== ========
+========= ======== ========
+Src / Dst Diff_dst Diff_src
+========= ======== ========
+f32       f32      f32
+f16       f16      f16
+bf16      bf16     bf16
+========= ======== ========
diff --git a/source/elements/oneDNN/source/graph/ops/Concat.rst b/source/elements/oneDNN/source/graph/ops/Concat.rst
@@ -69,10 +69,10 @@ Supported Data Types
 
 Concat operation supports the following data type combinations.
 
-==== ====
-Src  Dst
-==== ====
-f32  f32
-bf16 bf16
-f16  f16
-==== ====
+===== ====
+Src_i Dst
+===== ====
+f32   f32
+bf16  bf16
+f16   f16
+===== ====
diff --git a/source/elements/oneDNN/source/graph/ops/Divide.rst b/source/elements/oneDNN/source/graph/ops/Divide.rst
@@ -69,10 +69,10 @@ Supported Data Types
 
 Divide operation supports the following data type combinations.
 
-======== =========
-Diff_dst Diff_bias
-======== =========
-f32      f32
-bf16     bf16
-f16      f16
-======== =========
+============= ====
+Src_0 / Src_1 Dst
+============= ====
+f32           f32
+bf16          bf16
+f16           f16
+============= ====
diff --git a/source/elements/oneDNN/source/graph/ops/EluBackward.rst b/source/elements/oneDNN/source/graph/ops/EluBackward.rst
@@ -25,7 +25,7 @@ Operation Attributes
 +----------------+-------------+------------+--------------+-------------+
 | |attr_use_dst| | If true,    | bool       | ``true``     | Optional    |
 |                | use         |            | (default),   |             |
-|                | \diffsrc    |            | ``false``    |             |
+|                | \dst        |            | ``false``    |             |
 |                | of Elu      |            |              |             |
 |                | operation   |            |              |             |
 |                | to          |            |              |             |
@@ -47,12 +47,12 @@ Inputs
 ======
 
 
-===== ============= ====================
-Index Argument Name Required or Optional
-===== ============= ====================
-0     ``src``       Required
-1     ``diff_dst``  Required
-===== ============= ====================
+===== ================= ====================
+Index Argument Name     Required or Optional
+===== ================= ====================
+0     ``src`` / ``dst`` Required
+1     ``diff_dst``      Required
+===== ================= ====================
 
 Outputs
 =======
@@ -70,10 +70,10 @@ Supported Data Types
 
 EluBackward operation supports the following data type combinations.
 
-==== ======== ========
-Src  Diff_dst Diff_src
-==== ======== ========
-f32  f32      f32
-f16  f16      f16
-bf16 bf16     bf16
-==== ======== ========
+========= ======== ========
+Src / Dst Diff_dst Diff_src
+========= ======== ========
+f32       f32      f32
+f16       f16      f16
+bf16      bf16     bf16
+========= ======== ========
diff --git a/source/elements/oneDNN/source/graph/ops/Interpolate.rst b/source/elements/oneDNN/source/graph/ops/Interpolate.rst
@@ -111,10 +111,10 @@ Supported Data Types
 
 Interpolate operation supports the following data type combinations.
 
-======= =====
-Src/Dst Sizes
-======= =====
-f32     s32
-bf16    s32
-f16     s3
-======= =====
+========= =====
+Src / Dst Sizes
+========= =====
+f32       s32
+bf16      s32
+f16       s32
+========= =====
diff --git a/source/elements/oneDNN/source/graph/ops/InterpolateBackward.rst b/source/elements/oneDNN/source/graph/ops/InterpolateBackward.rst
@@ -113,10 +113,10 @@ Supported Data Types
 InterpolateBackward operation supports the following data type
 combinations.
 
-===================== =====
-Src/Diff_dst/Diff_src Sizes
-===================== =====
-f32                   s32
-bf16                  s32
-f16                   s32
-===================== =====
+==== ======== ======== =====
+Src  Diff_dst Diff_src Sizes
+==== ======== ======== =====
+f32  f32      f32      s32
+bf16 bf16     bf16     s32
+f16  f16      f16      s32
+==== ======== ======== =====
diff --git a/source/elements/oneDNN/source/graph/ops/LogSoftmaxBackward.rst b/source/elements/oneDNN/source/graph/ops/LogSoftmaxBackward.rst
@@ -39,7 +39,7 @@ Inputs
 Index Argument Name Required or Optional
 ===== ============= ====================
 0     ``diff_dst``  Required
-1     ``src``       Required
+1     ``dst``       Required
 ===== ============= ====================
 
 Outputs
@@ -56,13 +56,12 @@ Index Argument Name Required or Optional
 Supported Data Types
 ********************
 
-LogSoftmaxBackward operation supports the following data type
-combinations.
+LogSoftmaxBackward operation supports the following data type combinations.
 
-==== ====
-Src  Dst
-==== ====
-f32  f32
-bf16 bf16
-f16  f16
-==== ====
+======== ==== ========
+Diff_dst Dst  Diff_src
+======== ==== ========
+f32      f32  f32
+bf16     bf16 bf16
+f16      f16  f16
+======== ==== ========
diff --git a/source/elements/oneDNN/source/graph/ops/Maximum.rst b/source/elements/oneDNN/source/graph/ops/Maximum.rst
@@ -69,10 +69,10 @@ Supported Data Types
 
 Maximum operation supports the following data type combinations.
 
-========= ===========
-Source0/1 Destination
-========= ===========
-f32       f32
-bf16      bf16
-f16       f16
-========= ===========
+============= ====
+Src_0 / Src_1 Dst
+============= ====
+f32           f32
+bf16          bf16
+f16           f16
+============= ====
diff --git a/source/elements/oneDNN/source/graph/ops/Minimum.rst b/source/elements/oneDNN/source/graph/ops/Minimum.rst
@@ -69,10 +69,10 @@ Supported Data Types
 
 Minimum operation supports the following data type combinations.
 
-========= ===========
-Source0/1 Destination
-========= ===========
-f32       f32
-bf16      bf16
-f16       f16
-========= ===========
+============= ====
+Src_0 / Src_1 Dst
+============= ====
+f32           f32
+bf16          bf16
+f16           f16
+============= ====
diff --git a/source/elements/oneDNN/source/graph/ops/Multiply.rst b/source/elements/oneDNN/source/graph/ops/Multiply.rst
@@ -70,10 +70,10 @@ Supported Data Types
 
 Multiply operation supports the following data type combinations.
 
-========= ===========
-Source0/1 Destination
-========= ===========
-f32       f32
-bf16      bf16
-f16       f16
-========= ===========
+============= ====
+Src_0 / Src_1 Dst
+============= ====
+f32           f32
+bf16          bf16
+f16           f16
+============= ====
diff --git a/source/elements/oneDNN/source/graph/ops/Pow.rst 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/ReLUBackward.rst b/source/elements/oneDNN/source/graph/ops/ReLUBackward.rst
@@ -63,10 +63,10 @@ Supported Data Types
 
 ReLUBackward operation supports the following data type combinations.
 
-==== ======== ========
-Src  Diff_dst Diff_src
-==== ======== ========
-f32  f32      f32
-f16  f16      f16
-bf16 bf16     bf16
-==== ======== ========
+========= ======== ========
+Src / Dst Diff_dst Diff_src
+========= ======== ========
+f32       f32      f32
+f16       f16      f16
+bf16      bf16     bf16
+========= ======== ========
diff --git a/source/elements/oneDNN/source/graph/ops/Reciprocal.rst b/source/elements/oneDNN/source/graph/ops/Reciprocal.rst
@@ -55,10 +55,10 @@ Supported Data Types
 
 Reciprocal operation supports the following data type combinations.
 
-====== ===========
-Source Destination
-====== ===========
-f32    f32
-bf16   bf16
-f16    f16
-====== ===========
+==== ====
+Src  Dst
+==== ====
+f32  f32
+bf16 bf16
+f16  f16
+==== ====
diff --git a/source/elements/oneDNN/source/graph/ops/ReduceL1.rst b/source/elements/oneDNN/source/graph/ops/ReduceL1.rst
@@ -126,10 +126,10 @@ Supported Data Types
 
 ReduceL1 operation supports the following data type combinations.
 
-================== ====
-Source/Destination Axes
-================== ====
-f32                s32
-bf16               s32
-f16                s32
-================== ====
+==== ==== ====
+Src  Dst  Axes
+==== ==== ====
+f32  f32  s32
+bf16 bf16 s32
+f16  f16  s32
+==== ==== ====