函数: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。模型构建的详细说明请参见模型构建。 |
父主题: aclmdlDesc