Feature Scope
Data Types
src Data Type |
dst Data Type |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
Data Layout
The KDNN Sum operator supports the following data layout:
- The data dimension can be 1D to 5D.
- The N input tensors and output tensors must have the same dimension and data layout. For details, see the following table.
Tensor Dimension |
src Data Layout |
dst Data Layout |
|---|---|---|
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 |
Parent topic: Sum