设置matmul计算时的单次计算的形状singleM、singleN、singleK,单位为元素个数。
int32_t SetShape(int32_t singleM, int32_t singleN, int32_t singleK);
参数名 |
输入/输出 |
描述 |
---|---|---|
singleM |
输入 |
设置单次计算的singleM大小,单位为元素。 |
singleN |
输入 |
设置单次计算的singleN大小,单位为元素。 |
singleK |
输入 |
设置单次计算的singleK小,单位为元素。 |
-1表示设置失败; 0表示设置成功。
无