特性范围
传播方向
传播方向 |
具体类别 |
|---|---|
Forward |
dnnl_forward_training dnnl_forward_inference |
Backward |
dnnl_backward_data |
数据类型
输入src数据类型 |
输出dst数据类型 |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
维度与数据排布
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 |
父主题： Softmax