OP_OUTSHAPE
Function Usage
Stores the aclTensor of the output shape of an operator, for example, the NoneZero operator.
Prototype
OP_OUTSHAPE(x...)
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
x... |
Input |
The parameter consists of two parts. The first parameter is the aclTensor that stores the output tensor shape, and the second parameter is the index of the tensor whose output shape needs to be updated. |
Constraints
None
Example
1 2 | // The operator stores the shape of the output tensor in outShapeTensor to update the shape of the output tensor whose idx is 0. OP_OUTSHAPE({outShapeTensor, 0}); |
Parent topic: Common Macros and Classes