文档管理中心

GetComputeNodeInputNum

函数功能

获取算子的输入个数。

函数原型

收起
自动换行
深色代码主题
复制
  1. size_t GetComputeNodeInputNum() const

参数说明

无

返回值

算子的输入个数。

约束说明

无

调用示例

收起
自动换行
深色代码主题
复制
  1. // 假设已存在KernelContext *context
  2. auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
  3. for (size_t idx = 0; idx < extend_context->GetComputeNodeInputNum(); ++idx) {
  4. auto input_td = extend_context->GetInputDesc(idx);
  5. // ...
  6. }