函数: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,卷积的权重格式。

约束说明

注意事项