SetOrgShape
功能说明
设置matmul计算时的原始完整的形状M、N、K，单位为元素个数。
函数原型
int32_t SetOrgShape(int32_t orgM, int32_t orgN, int32_t orgK)
int32_t SetOrgShape(int32_t orgM, int32_t orgN, int32_t orgKa, int32_t orgKb)
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
orgM |
输入 |
设置原始完整的形状M大小，单位为元素。 |
orgN |
输入 |
设置原始完整的形状N大小，单位为元素。 |
orgK |
输入 |
设置原始完整的形状K大小，单位为元素。原始完整形状Ka=Kb时可设置。 |
orgKa |
输入 |
设置矩阵A原始完整的形状Ka大小，单位为元素。 |
orgKb |
输入 |
设置矩阵B原始完整的形状Kb大小，单位为元素。 |
返回值
-1表示设置失败； 0表示设置成功。
注意事项
无
父主题： Matmul Tiling类