文档
注册

函数: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类型数据的指针地址。

需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。

返回值说明

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

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

    batch = {
    "batchCount": int,#模型中支持的batch分档数
    "batch":[xx, xx,]  # 模型中支持的具体分档
    }
  • batchCount等于0时,表示不支持设置档位信息,以模型中的档位为准。

ret:int,错误码。

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

约束说明

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

当前产品无相关内容

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