SetBatchNum
功能说明
调用IterateBatch接口之前,需要在host侧Tiling实现中通过本接口设置多Batch计算的最大Batch数。
函数原型
int32_t SetBatchNum(int32_t batch);
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
batch |
输入 |
多Batch计算的最大Batch数。 |
返回值
-1表示设置失败; 0表示设置成功。
注意事项
无
调用IterateBatch接口之前,需要在host侧Tiling实现中通过本接口设置多Batch计算的最大Batch数。
int32_t SetBatchNum(int32_t batch);
参数名 |
输入/输出 |
描述 |
---|---|---|
batch |
输入 |
多Batch计算的最大Batch数。 |
-1表示设置失败; 0表示设置成功。
无