更新时间:2021-03-18 GMT+08:00
分享

aclopSetKernelWorkspaceSizes

函数功能

动态Shape场景下,设置算子Workspace参数。

约束说明

为非必须接口,根据算子情况可选。

函数原型

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

参数说明

参数名

输入/输出

说明

kernelDesc

输入

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

typedef struct aclopKernelDesc aclopKernelDesc

numWorkspaces

输入

Workspaces个数。

workspaceSizes

输入

Workspaces大小的数组地址。

返回值说明

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

分享:

    相关文档

    相关产品

close