GetComputeNodeOutputNum
Function Usage
Obtains the number of outputs of an operator.
Prototype
1 | size_t GetComputeNodeOutputNum() const |
Parameters
None
Returns
Number of outputs of an operator.
Constraints
None
Examples
1 2 3 4 5 6 | // Assume that KernelContext *context already exists. auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context); for (size_t idx = 0; idx < extend_context->GetComputeNodeOutputNum(); ++idx) { auto input_td = extend_context->GetOutputDesc(idx); ... } |
Parent topic: ExtendedKernelContext