SetBatchNum
功能说明
在不改变tiling的情况下,重新设置多Batch计算的Batch数。
函数原型
__aicore__ inline void SetBatchNum(int32_t batchA, int32_t batchB)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
batchA |
输入 |
设置的一次计算的A矩阵batch数。 |
batchB |
输入 |
设置的一次计算的B矩阵batch数。 |
返回值
无
注意事项
无
父主题: Matmul