下载
中文
注册

SetBatchNum

功能说明

调用IterateBatch接口之前,需要在host侧Tiling实现中通过本接口设置多Batch计算的最大Batch数。

函数原型

int32_t SetBatchNum(int32_t batch);

参数说明

表1 参数说明

参数名

输入/输出

描述

batch

输入

多Batch计算的最大Batch数。

返回值

-1表示设置失败; 0表示设置成功。

注意事项