Skip to content

Commit da2fcac

Browse files
authored
[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
1 parent e0eabad commit da2fcac

32 files changed

Lines changed: 386 additions & 224 deletions

source/elements/oneDNN/source/graph/data_model.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@ it's suggested to refer to the definition page of each operation.
3939

4040
u8
4141
8-bit unsigned integer.
42+
43+
boolean
44+
Boolean data type. Size is C++ implementation defined.
4245

4346

4447
oneDNN Graph supports both public layout and |opaque| layout. When the

source/elements/oneDNN/source/graph/ops/Add.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -68,10 +68,10 @@ Supported Data Types
6868

6969
Add operation supports the following data type combinations.
7070

71-
========= ===========
72-
Source0/1 Destination
73-
========= ===========
74-
f32 f32
75-
bf16 bf16
76-
f16 f16
77-
========= ===========
71+
============= ====
72+
Src_0 / Src_1 Dst
73+
============= ====
74+
f32 f32
75+
bf16 bf16
76+
f16 f16
77+
============= ====

source/elements/oneDNN/source/graph/ops/BatchNormTrainingBackward.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -78,10 +78,10 @@ Supported Data Types
7878
BatchNormTrainingBackward operation supports the following data type
7979
combinations.
8080

81-
==================== ================================================
82-
Src / Diff_dst / Dst Mean / Variance / Gamma / Diff_gamma / Diff_beta
83-
==================== ================================================
84-
f32 f32
85-
bf16 f32, bf16
86-
f16 f32
87-
==================== ================================================
81+
========================= ================================================
82+
Src / Diff_dst / Diff_src Mean / Variance / Gamma / Diff_gamma / Diff_beta
83+
========================= ================================================
84+
f32 f32
85+
bf16 f32, bf16
86+
f16 f32
87+
========================= ================================================

source/elements/oneDNN/source/graph/ops/ClampBackward.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -93,10 +93,10 @@ Supported Data Types
9393

9494
ClampBackward operation supports the following data type combinations.
9595

96-
==== ======== ========
97-
Src Diff_dst Diff_src
98-
==== ======== ========
99-
f32 f32 f32
100-
f16 f16 f16
101-
bf16 bf16 bf16
102-
==== ======== ========
96+
========= ======== ========
97+
Src / Dst Diff_dst Diff_src
98+
========= ======== ========
99+
f32 f32 f32
100+
f16 f16 f16
101+
bf16 bf16 bf16
102+
========= ======== ========

source/elements/oneDNN/source/graph/ops/Concat.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,10 @@ Supported Data Types
6969

7070
Concat operation supports the following data type combinations.
7171

72-
==== ====
73-
Src Dst
74-
==== ====
75-
f32 f32
76-
bf16 bf16
77-
f16 f16
78-
==== ====
72+
===== ====
73+
Src_i Dst
74+
===== ====
75+
f32 f32
76+
bf16 bf16
77+
f16 f16
78+
===== ====

source/elements/oneDNN/source/graph/ops/Divide.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -69,10 +69,10 @@ Supported Data Types
6969

7070
Divide operation supports the following data type combinations.
7171

72-
======== =========
73-
Diff_dst Diff_bias
74-
======== =========
75-
f32 f32
76-
bf16 bf16
77-
f16 f16
78-
======== =========
72+
============= ====
73+
Src_0 / Src_1 Dst
74+
============= ====
75+
f32 f32
76+
bf16 bf16
77+
f16 f16
78+
============= ====

source/elements/oneDNN/source/graph/ops/EluBackward.rst

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ Operation Attributes
2525
+----------------+-------------+------------+--------------+-------------+
2626
| |attr_use_dst| | If true, | bool | ``true`` | Optional |
2727
| | use | | (default), | |
28-
| | \diffsrc | | ``false`` | |
28+
| | \dst | | ``false`` | |
2929
| | of Elu | | | |
3030
| | operation | | | |
3131
| | to | | | |
@@ -47,12 +47,12 @@ Inputs
4747
======
4848

4949

50-
===== ============= ====================
51-
Index Argument Name Required or Optional
52-
===== ============= ====================
53-
0 ``src`` Required
54-
1 ``diff_dst`` Required
55-
===== ============= ====================
50+
===== ================= ====================
51+
Index Argument Name Required or Optional
52+
===== ================= ====================
53+
0 ``src`` / ``dst`` Required
54+
1 ``diff_dst`` Required
55+
===== ================= ====================
5656

5757
Outputs
5858
=======
@@ -70,10 +70,10 @@ Supported Data Types
7070

7171
EluBackward operation supports the following data type combinations.
7272

73-
==== ======== ========
74-
Src Diff_dst Diff_src
75-
==== ======== ========
76-
f32 f32 f32
77-
f16 f16 f16
78-
bf16 bf16 bf16
79-
==== ======== ========
73+
========= ======== ========
74+
Src / Dst Diff_dst Diff_src
75+
========= ======== ========
76+
f32 f32 f32
77+
f16 f16 f16
78+
bf16 bf16 bf16
79+
========= ======== ========

source/elements/oneDNN/source/graph/ops/Interpolate.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,10 @@ Supported Data Types
111111

112112
Interpolate operation supports the following data type combinations.
113113

114-
======= =====
115-
Src/Dst Sizes
116-
======= =====
117-
f32 s32
118-
bf16 s32
119-
f16 s3
120-
======= =====
114+
========= =====
115+
Src / Dst Sizes
116+
========= =====
117+
f32 s32
118+
bf16 s32
119+
f16 s32
120+
========= =====

source/elements/oneDNN/source/graph/ops/InterpolateBackward.rst

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -113,10 +113,10 @@ Supported Data Types
113113
InterpolateBackward operation supports the following data type
114114
combinations.
115115

116-
===================== =====
117-
Src/Diff_dst/Diff_src Sizes
118-
===================== =====
119-
f32 s32
120-
bf16 s32
121-
f16 s32
122-
===================== =====
116+
==== ======== ======== =====
117+
Src Diff_dst Diff_src Sizes
118+
==== ======== ======== =====
119+
f32 f32 f32 s32
120+
bf16 bf16 bf16 s32
121+
f16 f16 f16 s32
122+
==== ======== ======== =====

source/elements/oneDNN/source/graph/ops/LogSoftmaxBackward.rst

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ Inputs
3939
Index Argument Name Required or Optional
4040
===== ============= ====================
4141
0 ``diff_dst`` Required
42-
1 ``src`` Required
42+
1 ``dst`` Required
4343
===== ============= ====================
4444

4545
Outputs
@@ -56,13 +56,12 @@ Index Argument Name Required or Optional
5656
Supported Data Types
5757
********************
5858

59-
LogSoftmaxBackward operation supports the following data type
60-
combinations.
59+
LogSoftmaxBackward operation supports the following data type combinations.
6160

62-
==== ====
63-
Src Dst
64-
==== ====
65-
f32 f32
66-
bf16 bf16
67-
f16 f16
68-
==== ====
61+
======== ==== ========
62+
Diff_dst Dst Diff_src
63+
======== ==== ========
64+
f32 f32 f32
65+
bf16 bf16 bf16
66+
f16 f16 f16
67+
======== ==== ========

0 commit comments

Comments
 (0)