Function: set_kernel_workspace_sizes
C Prototype |
aclError aclopSetKernelWorkspaceSizes(aclopKernelDesc *kernelDesc, int numWorkspaces, size_t *workspaceSizes) |
|---|---|
Python Function |
ret = acl.op.set_kernel_workspace_sizes(kernel_desc, num_workspaces, workspace_sizes) |
Function Usage |
Sets the workspace parameters of an operator in the dynamic shape scenario. |
Input Description |
kernel_desc: int, Kernel description buffer, pointer address of the aclopKernelDesc type. num_workspaces: int, number of workspaces. workspace_sizes: array address of the workspace sizes. |
Return Value |
ret: int, error code.
|
Restrictions |
This is an optional interface and can be selected as required. |
Parent topic: Operator Building