Rate This Document
Findability
Accuracy
Completeness
Readability

Feature Scope

Propagation Directions and Data Types

Operation

Data Type

Supported Propagation Directions

abs

f32

Forward:

  • dnnl_forward_training
  • dnnl_forward_inference

exp

f32

log

f32

sqrt

f32

round

f32

tanh

f32

relu

f32/f16/bf16

  • Forward:
    • dnnl_forward_training
    • dnnl_forward_inference
  • Backward: dnnl_backward_data

logistic

f32/f16/bf16

linear

f32/f16/bf16

Dimensions and Data Layout

The Eltwise operator in KDNN supports 1D to 5D and sequential data layout.

Table 1 Mapping between each tensor dimension and parameter 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