鲲鹏社区首页
中文
注册
开发者
我要评分
获取效率
正确性
完整性
易理解
在线提单
论坛求助

特性范围

数据类型

传播方向

src数据类型

dst数据类型

alg类型

forward/backward

f32

f32

max,avg_p,avg_np

forward/backward

f16

f16

max,avg_p,avg_np

forward/backward

bf16

bf16

max,avg_p,avg_np

forward

s8

s8

max,avg_p,avg_np

forward

u8

u8

max,avg_p,avg_np

forward

s32

s32

max,avg_p,avg_np

数据排布

支持1D至5D Tensor维度,具体数据排布如下。

Tensor维度

src数据排布

dst数据排布

1D Tensor

a

a

2D Tensor

ab、ba

ab、ba

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