特性范围
数据类型
src数据类型 |
dst数据类型 |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
数据排布
KAIL_DNN Sum算子支持以下数据排布：
- 数据维度支持1D-5D。
- N个输入Tensor与输出Tensor需为相同维度、相同数据排布，具体数据排布支持情况见下表。
Tensor维度 |
src数据排布 |
dst数据排布 |
|---|---|---|
1D |
dnnl_a |
dnnl_a |
2D |
dnnl_ab |
dnnl_ab |
3D |
dnnl_abc |
dnnl_abc |
dnnl_acb |
dnnl_acb |
|
4D |
dnnl_abcd |
dnnl_abcd |
dnnl_acdb |
dnnl_acdb |
|
5D |
dnnl_abcde |
dnnl_abcde |
dnnl_acdeb |
dnnl_acdeb |
父主题： Sum