SetShape
功能说明
设置matmul计算时的单次计算的形状singleM、singleN、singleK,单位为元素个数。
函数原型
int32_t SetShape(int32_t singleM, int32_t singleN, int32_t singleK)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
singleM |
输入 |
设置单次计算的singleM大小,单位为元素。 |
singleN |
输入 |
设置单次计算的singleN大小,单位为元素。 |
singleK |
输入 |
设置单次计算的singleK小,单位为元素。 |
返回值
-1表示设置失败; 0表示设置成功。
注意事项
无
父主题: Matmul Tiling类