特性范围
传播方向
传播方向 |
具体类别 |
|---|---|
Forward |
dnnl_forward_training dnnl_forward_inference |
Backward |
dnnl_backward_data dnnl_backward_weights |
数据类型
输入src数据类型 |
权重weight数据类型 |
输出dst数据类型 |
偏置bias数据类型 |
|---|---|---|---|
f32 |
f32 |
f32 |
f32/无 |
f16 |
f16 |
f16 |
f16/无 |
bf16 |
bf16 |
bf16 |
bf16/无 |
f16 |
f16 |
f32 |
f32/无 |
bf16 |
bf16 |
f32 |
f32/无 |
输入src数据类型 |
权重weight数据类型 |
输出dst数据类型 |
|---|---|---|
f32 |
f32 |
f32 |
f16 |
f16 |
f16 |
bf16 |
bf16 |
bf16 |
f32 |
f16 |
f16 |
f32 |
bf16 |
bf16 |
输入src数据类型 |
权重weight数据类型 |
输出dst数据类型 |
偏置bias数据类型 |
|---|---|---|---|
f32 |
f32 |
f32 |
f32/无 |
f16 |
f16 |
f16 |
f16/无 |
bf16 |
bf16 |
bf16 |
bf16/无 |
f16 |
f32 |
f16 |
f32/无 |
bf16 |
f32 |
bf16 |
f32/无 |
父主题： Inner Product