SetSingleShape
功能说明
设置Matmul单核计算的形状singleMIn,singleNIn,singleKIn,单位为元素个数。
函数原型
virtual int32_t SetSingleShape(int32_t singleMIn = -1, int32_t singleNIn = -1, int32_t singleKIn = -1);
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
singleMIn |
输入 |
设置的singleMIn大小,单位为元素,默认值为-1。 |
singleNIn |
输入 |
设置的singleNIn大小,单位为元素,默认值为-1。 |
singleKIn |
输入 |
设置的singleKIn大小,单位为元素,默认值为-1。 |
返回值
-1表示设置失败; 0表示设置成功。
注意事项
无
父主题: Matmul Tiling类