Documentation
Rate This Document
Findability
Accuracy
Completeness
Readability
SubmitClose

Feature Scope

Propagation Directions

Propagation Direction

Specific Category

Forward

dnnl_forward_training

dnnl_forward_inference

Backward

dnnl_backward_data

dnnl_backward_weights

Data Types

Table 1 Parameter data types of the forward direction

Input Data (src) Type

Weight Data Type

Output Data (dst) Type

Bias Data Type

f32

f32

f32

f32/none

f16

f16

f16

f16/none

bf16

bf16

bf16

bf16/none

f16

f16

f32

f32/none

bf16

bf16

f32

f32/none
Table 2 Parameter data types of the backward direction (dnnl_backward_data category)

Input Data (src) Type

Weight Data Type

Output Data (dst) Type

f32

f32

f32

f16

f16

f16

bf16

bf16

bf16

f32

f16

f16

f32

bf16

bf16
Table 3 Parameter data types of the backward direction (dnnl_backward_weights category)

Input Data (src) Type

Weight Data Type

Output Data (dst) Type

Bias Data Type

f32

f32

f32

f32/none

f16

f16

f16

f16/none

bf16

bf16

bf16

bf16/none

f16

f32

f16

f32/none

bf16

f32

bf16

f32/none
Parent topic: Inner Product