文档
注册

aclopSetKernelWorkspaceSizes

函数功能

动态Shape场景下,设置算子Workspace参数。为非必须接口,根据算子情况可选。

函数原型

aclError aclopSetKernelWorkspaceSizes(aclopKernelDesc *kernelDesc, int numWorkspaces, size_t *workspaceSizes)

参数说明

参数名

输入/输出

说明

kernelDesc

输入

Kernel描述缓存,aclopKernelDesc类型的指针。

typedef struct aclopKernelDesc aclopKernelDesc

numWorkspaces

输入

Workspaces个数。

workspaceSizes

输入

Workspaces大小的数组地址指针。

返回值说明

返回0表示成功,返回其它值表示失败。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词