aclmdlGetInputDynamicGearCount
产品支持情况
功能说明
根据模型描述信息获取模型的输入所支持的动态维度档位数。
函数原型
aclError aclmdlGetInputDynamicGearCount(const aclmdlDesc *modelDesc, size_t index, size_t *gearCount)
参数说明
|
需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。 |
||
|
例如,模型的输入tensor是4维,在模型转换时通过--dynamic_dims参数设置的分档为“1,3,224,224;2,3,224,224;1,3,256,256”,那么通过该接口获取的动态维度档位数为3。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。
约束说明
如果模型构建时没有设置动态维度的分档,那么通过该接口获取的动态维度档位数为0。