获取算子的输入个数。
- size_t GetComputeNodeInputNum() const
无
算子的输入个数。
无
- // 假设已存在KernelContext *context
- auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
- for (size_t idx = 0; idx < extend_context->GetComputeNodeInputNum(); ++idx) {
- auto input_td = extend_context->GetInputDesc(idx);
- // ...
- }