下载
中文
注册

函数:get_input_dynamic_gear_count

C函数原型

aclError aclmdlGetInputDynamicGearCount(const aclmdlDesc *modelDesc, size_t index, size_t *gearCount)

Python函数

gear_count, ret = acl.mdl.get_input_dynamic_gear_count(model_desc, index)

函数功能

根据模型描述信息获取模型的输入所支持的动态维度档位数。同步接口。

输入说明

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

index:int,预留参数,当前未使用,固定设置为-1。

返回值说明

gear_count:int,

动态维度档位数。

例如,模型的输入tensor是4维,在模型转换时通过--dynamic_dims参数设置的分档为“1,3,224,224;2,3,224,224;1,3,256,256”,那么通过该接口获取的动态维度档位数为3。

ret:int,错误码。
  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

如果模型构建时没有设置动态维度的分档,那么通过该接口获取的动态维度档位数为0。模型构建的详细说明请参见模型构建