OP_WORKSPACE
Function Usage
Encapsulates the workspace parameter explicitly specified by the operator.
Prototype
OP_WORKSPACE(x...)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
x... |
Input |
Workspace parameter of the operator, indicating the device memory required for operator computation. The aclTensor or aclTensorList type is supported. |
Constraints
None
Example
1 2 | // Encapsulate the workspace parameter a of the operator. OP_WORKSPACE(a); |
Parent topic: Common Macros and Classes