函数:get_input_dynamic_gear_count
产品支持情况
产品 |
是否支持 |
---|---|
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
根据模型描述信息获取模型的输入所支持的动态维度档位数。
函数原型
- C函数原型
1
aclError aclmdlGetInputDynamicGearCount(const aclmdlDesc *modelDesc, size_t index, size_t *gearCount)
- python函数
1
gear_count, ret = acl.mdl.get_input_dynamic_gear_count(model_desc, index)
参数说明
参数名 |
说明 |
---|---|
model_desc |
int,aclmdlDesc类型模型描述信息数据的指针地址。 需提前调用acl.mdl.create_desc接口创建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,错误码。
|
父主题: aclmdlDesc