昇腾社区首页
中文
注册
开发者
下载

GetInferenceContext

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件和库文件

  • 头文件:#include <graph/operator.h>
  • 库文件:libgraph.so

功能说明

获取当前算子传递infershape推导所需要的关联信息,比如前面算子的shape和DataType信息。

函数原型

1
InferenceContextPtr GetInferenceContext() const

参数说明

无。

返回值说明

返回当前operator的推理上下文。

InferenceContextPtr是指向InferenceContext类的指针的别名:

1
using InferenceContextPtr = std::shared_ptr<InferenceContext>;

异常处理

无。

约束说明

无。