特性范围
数据类型
src数据类型 |
dst数据类型 |
|---|---|
f32 |
f32 |
f16 |
f16 |
bf16 |
bf16 |
数据排布
支持1D至5D Tensor维度,具体数据排布如下。
Tensor维度 |
src数据排布 |
dst数据排布 |
|---|---|---|
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 |