GetComputeNodeInputNum

Description

Obtains the number of inputs of an operator.

Prototype

size_t GetComputeNodeInputNum() const

Parameters

None

Returns

Number of inputs of an operator.

Restrictions

None

Example

// 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);
  ...
}