文档
注册

函数:get_dynamic_batch

C函数原型

aclError aclmdlGetDynamicBatch(const aclmdlDesc *modelDesc, aclmdlBatch *batch)

Python函数

batch, ret = acl.mdl.get_dynamic_batch(model_desc)

函数功能

获取模型支持的动态Batch信息。同步接口。

输入说明

model_desc:int,aclmdlDesc类型数据的指针地址

返回值说明

batch:dict,获取的模型支持的动态Batch信息

  • batch = {

    "batchCount": int,#模型中支持的batch分档数

    "batch":[xx, xx,] # 模型中支持的具体分档

    }

    模型中支持的最大batch数为128

  • batchCount等于0时,表示不支持设置档位信息,以模型中的档位为准。

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词