接口功能: 在Matmul算子ND格式输入下,计算需要申请的weight的大小,该接口仅仅用于判断对weight Tensor进行预处理需要使用多少size才可使Matmul算子执行性能最优。 例如输入【510, 510】:该函数出于性能角度考虑,会将shape变化为【512,512】,因此函数会将引用输入修改为262144。
计算公式:
[object Object]
- 确定性计算:
- aclnnCalculateMatmulWeightSize默认确定性实现。
[object Object]
接口功能: 在Matmul算子ND格式输入下,计算需要申请的weight的大小,该接口仅仅用于判断对weight Tensor进行预处理需要使用多少size才可使Matmul算子执行性能最优。 例如输入【510, 510】:该函数出于性能角度考虑,会将shape变化为【512,512】,因此函数会将引用输入修改为262144。
计算公式:
Float16/Bfloat16:result=i∈(0,3]∏Align(tensorShape[i],16) INT8:result=Align(Shapesize[0],16)∗Align(Shapesize[1],32)