调用IterateBatch接口之前,需要在host侧Tiling实现中通过本接口设置多Batch计算的最大Batch数。
int32_t SetBatchNum(int32_t batch)
参数名
输入/输出
描述
batch
输入
多Batch计算的最大Batch数。
-1表示设置失败; 0表示设置成功。
无