Feature Scope
Propagation Directions
Propagation Direction |
Specific Category |
|---|---|
Forward |
dnnl_forward_training dnnl_forward_inference |
Backward |
dnnl_backward_data |
Data Types
Input Data (src) Type |
Output Data (dst) Type |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
Dimensions and 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: Softmax