文档管理中心

GetIrInputInstanceInfo

函数功能

根据算子原型定义中的输入索引获取对应输入的实例化信息。

函数原型

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

参数说明

展开
参数 输入/输出 说明
ir_index 输入 算子IR原型定义中的输入索引,从0开始计数。

返回值

指定输入的实例化信息。

关于AnchorInstanceInfo的定义,请参见AnchorInstanceInfo

约束说明

调用示例

收起
自动换行
深色代码主题
复制
  1. // 假设已存在KernelContext *context
  2. auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
  3. for (size_t idx = 0; idx < extend_context->GetComputeNodeInfo()->GetIrInputsNum(); ++idx) {
  4. auto input_td = extend_context->GetIrInputInstanceInfo(idx);
  5. // ...
  6. }
在 开发与测试 开放能力API 中进行搜索
请输入您想要搜索的关键词