Feature Scope
Propagation Directions and Data Types
Operation |
Data Type |
Supported Propagation Directions |
|---|---|---|
abs |
f32 |
Forward:
|
exp |
f32 |
|
log |
f32 |
|
sqrt |
f32 |
|
round |
f32 |
|
tanh |
f32 |
|
relu |
f32/f16/bf16 |
|
logistic |
f32/f16/bf16 |
|
linear |
f32/f16/bf16 |
Dimensions and Data Layout
The Eltwise operator in KDNN supports 1D to 5D and sequential data layout.
Tensor Dimension |
Input Tensor (src) Data Layout |
Output Tensor (dst) Data Layout |
|---|---|---|
1D |
dnnl_a |
dnnl_a |
2D |
dnnl_ab |
dnnl_ab |
3D |
dnnl_abc |
dnnl_abc |
4D |
dnnl_abcd |
dnnl_abcd |
5D |
dnnl_abcde |
dnnl_abcde |
Parent topic: Eltwise