函数:set_kernel_workspace_sizes
产品支持情况
产品  | 
是否支持  | 
|---|---|
√  | 
|
√  | 
|
√  | 
|
√  | 
|
√  | 
|
√  | 
使用说明
C函数原型  | 
aclError aclopSetKernelWorkspaceSizes(aclopKernelDesc *kernelDesc, int numWorkspaces, size_t *workspaceSizes)  | 
|---|---|
Python函数  | 
ret = acl.op.set_kernel_workspace_sizes(kernel_desc, num_workspaces, workspace_sizes)  | 
函数功能  | 
动态Shape场景下,设置算子Workspace参数。  | 
输入说明  | 
kernel_desc:int,Kernel描述缓存,aclopKernelDesc类型的指针地址。 num_workspaces:int,Workspaces个数。 workspace_sizes:int,Workspaces大小的数组地址。  | 
返回值说明  | 
ret:int,错误码。 
  | 
约束说明  | 
为非必须接口,根据算子情况可选。  | 
父主题: 单算子模型执行