函数: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

约束说明

注意事项