特性范围
数据类型
src数据类型 |
weight数据类型 |
dst数据类型 |
bias数据类型 |
|---|---|---|---|
f32 |
f32 |
f32 |
f32 |
f16 |
f16 |
f16 |
f16 |
bf16 |
bf16 |
bf16 |
bf16 |
f16 |
f16 |
f32 |
f32/f16 |
bf16 |
bf16 |
f32 |
bf16/f32 |
数据排布
Tensor维度 |
src数据排布 |
weight数据排布 |
dst数据排布 |
|---|---|---|---|
2D |
dnnl_ab/dnnl_ba |
dnnl_ab/dnnl_ba |
dnnl_ab/dnnl_ba |
3D |
dnnl_abc/dnnl_acb |
dnnl_abc/dnnl_acb |
dnnl_abc/dnnl_acb |
4D |
dnnl_abcd/dnnl_abdc |
dnnl_abcd/dnnl_abdc |
dnnl_abcd/dnnl_abdc |
5D |
dnnl_abcde/dnnl_abced |
dnnl_abcde/dnnl_abced |
dnnl_abcde/dnnl_abced |
父主题： Matmul