跳到主要内容

aclmdlGetDescFromMem

aclmdlGetDescFromMem

函数功能

从内存获取该模型的模型描述信息。同步接口。

Atlas 200/500 A2推理产品不支持该接口。

函数原型

aclError aclmdlGetDescFromMem(aclmdlDesc *modelDesc, const void *model, size_t modelSize)

约束说明

通过本接口获取到的模型描述信息,无法应用于aclmdlGetOpAttr接口。

参数说明

参数名输入/输出说明
modelDesc输出aclmdlDesc类型的指针。 需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。
model输入存放模型数据的内存地址指针。
modelSize输入内存中的模型数据长度,单位Byte。

返回值说明

返回0表示成功,返回非0表示失败。

父主题: aclmdlDesc

aclCreateTensorDesc

函数功能

创建aclTensorDesc类型的数据,该数据类型用于描述tensor的数据类型、shape、format等信息。同步接口。

如需销毁aclTensorDesc类型的数据,请参见aclDestroyTensorDesc

函数原型

aclTensorDesc *aclCreateTensorDesc(aclDataType dataType,

int numDims,

const int64_t *dims,

aclFormat format)

参数说明

参数名输入/输出说明
dataType输入tensor描述的数据类型。
numDims输入tensor描述shape的维度个数。 numDims存在以下约束:
- numDims必须大于或等于0;
- numDims>0时,numDims的值必须与dims数组的长度保持一致;
- numDims=0时,系统不使用dims数组值,dims参数值无效。
dims输入tensor描述维度大小的指针。 dims是一个数组,数组中的每个元素表示tensor中每个维度的大小,如果数组中某个元素的值为0,则为空tensor。 如果用户需要使用空tensor,则在申请内存时,内存大小最小为1Byte,以保障后续业务正常运行。
format输入tensor描述的format。

返回值说明

返回aclTensorDesc类型的指针。

父主题: aclTensorDesc

aclDestroyTensorDesc

函数功能

销毁aclTensorDesc类型的数据。同步接口。

函数原型

void aclDestroyTensorDesc(const aclTensorDesc *desc)

参数说明

参数名输入/输出说明
desc输入待销毁的aclTensorDesc类型的指针。

返回值说明

父主题: aclTensorDesc

aclGetTensorDescType

函数功能

获取tensor描述中的数据类型。同步接口。

函数原型

aclDataType aclGetTensorDescType(const aclTensorDesc *desc)

参数说明

参数名输入/输出说明
desc输入aclTensorDesc类型的指针。 需提前调用aclCreateTensorDesc接口创建aclTensorDesc类型。

返回值说明

返回指定tensor描述的数据类型。

父主题: aclTensorDesc

在线提单