Rate This Document
Findability
Accuracy
Completeness
Readability

Feature Scope

Data Types

Propagation Direction

src/dst Data Type

forward/backward

f32, bf16, f16

Data Layout

3D to 5D tensors are supported. The following data layout formats are supported. The layout of input and output tensors must be the same.

Tensor Dimension

src Data Layout

dst Data Layout

3D Tensor

abc,acb,bac,bca,cab,cba

abc,acb,bac,bca,cab,cba

4D Tensor

abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab

abcd,abdc,acbd,acdb,adbc,adcb,bacd,bcda,cdab,cdba,dcab

5D Tensor

abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab

abcde,abced,abdec,acbde,acdeb,adecb,bacde,bcdea,cdeab,cdeba,decab

Parameter Constraints

None