GetComputeNodeInputNum
Function Usage
Obtains the number of inputs of an operator.
Prototype
1 | size_t GetComputeNodeInputNum() const |
Parameters
None
Returns
Number of inputs 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->GetComputeNodeInputNum(); ++idx) { auto input_td = extend_context->GetInputDesc(idx); ... } |
Parent topic: ExtendedKernelContext