目录

GetComputeNodeInputNum

Function Description

Obtains the number of inputs of an operator.

Function Prototype

size_t GetComputeNodeInputNum() const

Parameters

None

Return Value

Number of inputs of an operator.

Restrictions

None

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