Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Merged
merged 3 commits into from
Jun 29, 2023
Merged

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

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
9f355d9
fix wrong argument names
wuxun-zhang May 19, 2023
2bf056f
add new select op and boolean data type
wuxun-zhang May 19, 2023
1469f61
add Pow op in onednn graph
wuxun-zhang Jun 29, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions source/elements/oneDNN/source/graph/data_model.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Add.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
============= ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/BatchNormTrainingBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
========================= ================================================
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/ClampBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
========= ======== ========
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Concat.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
===== ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Divide.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
============= ====
28 changes: 14 additions & 14 deletions source/elements/oneDNN/source/graph/ops/EluBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ Operation Attributes
+----------------+-------------+------------+--------------+-------------+
| |attr_use_dst| | If true, | bool | ``true`` | Optional |
| | use | | (default), | |
| | \diffsrc | | ``false`` | |
| | \dst | | ``false`` | |
| | of Elu | | | |
| | operation | | | |
| | to | | | |
Expand All @@ -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
=======
Expand All @@ -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
========= ======== ========
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Interpolate.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
========= =====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/InterpolateBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
==== ======== ======== =====
19 changes: 9 additions & 10 deletions source/elements/oneDNN/source/graph/ops/LogSoftmaxBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ Inputs
Index Argument Name Required or Optional
===== ============= ====================
0 ``diff_dst`` Required
1 ``src`` Required
1 ``dst`` Required
===== ============= ====================

Outputs
Expand All @@ -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
======== ==== ========
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Maximum.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
============= ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Minimum.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
============= ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Multiply.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
============= ====
63 changes: 63 additions & 0 deletions source/elements/oneDNN/source/graph/ops/Pow.rst
View file
Open in desktop
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
==== ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/ReLUBackward.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
========= ======== ========
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/Reciprocal.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
==== ====
14 changes: 7 additions & 7 deletions source/elements/oneDNN/source/graph/ops/ReduceL1.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
==== ==== ====
Loading