特性范围
传播方向及数据类型
Operation |
数据类型 |
支持的传播方向 |
|---|---|---|
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 |
维度与数据排布
KAIL_DNN支持的Eltwise类算子支持1D-5D，顺序数据排布。
Tensor维度 |
输入src Tensor数据排布 |
输出dst Tensor数据排布 |
|---|---|---|
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 |
父主题： Eltwise