函数:get_output_data_type
函数:get_input_format
| C函数原型 | aclFormat aclmdlGetInputFormat(const aclmdlDesc *modelDesc, size_t index) |
|---|---|
| Python函数 | format_info = acl.mdl.get_input_format(model_desc, index) |
| 函数功能 | 获取模型中指定输入的Format。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。 |
| 返回值说明 | format_info:int,返回指定输入的Format。 - -1:ACL_FORMAT_UNDEFINED,未知格式,默认值。 - 0:ACL_FORMAT_NCHW,NCHW格式。 - 1:ACL_FORMAT_NHWC,NHWC格式。 - 2:ACL_FORMAT_ND,表示支持任意格式,仅有Square、Tanh等这些单输入对自身处理的算子外,其他需要慎用。 - 3:ACL_FORMAT_NC1HWC0,华为自研的5维数据格式。 - 4:ACL_FORMAT_FRACTAL_Z,卷积的权重格式。 |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数:get_output_format
| C函数原型 | aclFormat aclmdlGetOutputFormat(const aclmdlDesc *modelDesc, size_t index) |
|---|---|
| Python函数 | format_info = acl.mdl.get_output_format(model_desc, index) |
| 函数功能 | 获取模型中指定输出的Format。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。 |
| 返回值说明 | format_info:int,返回指定输入的Format。 - -1:ACL_FORMAT_UNDEFINED,未知格式,默认值。 - 0:ACL_FORMAT_NCHW,NCHW格式。 - 1:ACL_FORMAT_NHWC,NHWC格式。 - 2:ACL_FORMAT_ND,表示支持任意格式,仅有Square、Tanh等这些单输入对自身处理的算子外其他需要慎用。 - 3:ACL_FORMAT_NC1HWC0,华为自研的5维数据格式。 - 4:ACL_FORMAT_FRACTAL_Z,卷积的权重格式。 |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数:get_input_data_type
| C函数原型 | aclDataType aclmdlGetInputDataType(const aclmdlDesc *modelDesc, size_t index) |
|---|---|
| Python函数 | data_type = acl.mdl.get_input_data_type(model_desc, index) |
| 函数功能 | 获取模型中指定输入的数据类型。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输入的大小,index值从0开始。 |
| 返回值说明 | data_type:int,返回指定输入的数据类型。 - -1:ACL_DT_UNDEFINED,未知数据类型,默认值。 - 0:ACL_FLOAT - 1:ACL_FLOAT16 - 2:ACL_INT8 - 3:ACL_INT32 - 4:ACL_UINT8 - 6:ACL_INT16 - 7:ACL_UINT16 - 8:ACL_UINT32 - 9:ACL_INT64 - 10:ACL_UINT64 - 11:ACL_DOUBLE - 12:ACL_BOOL |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数:get_output_data_type
| C函数原型 | aclDataType aclmdlGetOutputDataType(const aclmdlDesc *modelDesc, size_t index) |
|---|---|
| Python函数 | data_type = acl.mdl.get_output_data_type(model_desc, index) |
| 函数功能 | 根据模型描述信息获取模型中指定输出的数据类型。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 index:int,指定获取第几个输入的大小,index值从0开始。 |
| 返回值说明 | ata_type:int,返回指定输入的数据类型。 - -1:ACL_DT_UNDEFINED,未知数据类型,默认值。 - 0:ACL_FLOAT - 1:ACL_FLOAT16 - 2:ACL_INT8 - 3:ACL_INT32 - 4:ACL_UINT8 - 6:ACL_INT16 - 7:ACL_UINT16 - 8:ACL_UINT32 - 9:ACL_INT64 - 10:ACL_UINT64 - 11:ACL_DOUBLE - 12:ACL_BOOL |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数:get_input_index_by_name
| C函数原型 | aclError aclmdlGetInputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index) |
|---|---|
| Python函数 | index, ret = acl.mdl.get_input_index_by_name(model_desc, name) |
| 函数功能 | 根据模型中的输入名称获取对应输入的索引编号。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 name:str,指定输入名称。 |
| 返回值说明 | index:int,获取对应输入的索引编号。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数:get_output_index_by_name
| C函数原型 | aclError aclmdlGetOutputIndexByName(const aclmdlDesc *modelDesc, const char *name, size_t *index) |
|---|---|
| Python函数 | index, ret = acl.mdl.get_output_index_by_name(model_desc, name) |
| 函数功能 | 根据模型中的输出名称获取对应输出的索引编号。同步接口。 |
| 输入说明 | model_desc:int,aclmdlDesc类型数据的指针地址。 name:str,指定输入名称。 |
| 返回值说明 | index:int,获取对应输入的索引编号。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: aclmdlDesc
函数: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类型数据的指针地址。 |
| 返回值说明 | batch:dict,获取的模型支持的动态Batch信息。 - batch = { "batchCount": int,#模型中支持的batch分档数 "batch":[xx, xx,] # 模型中支持的具体分档 } 模型中支持的最大batch数为128。 - batchCount等于0时,表示不支持设置档位信息,以模型中的档位为准。 ret:int,错误码。 - 返回0表示成功。 - 返回其它值表示失败。 |
| 约束说明 | 无 |
父主题: aclmdlDesc
在线提单