GetInferenceContext

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

头文件

#include <graph/ct_infer_shape_context.h>

功能说明

获取InferenceContext指针。

函数原型

ge::InferenceContext *GetInferenceContext() const

参数说明

无

返回值说明

输出InferenceContext指针。

关于InferenceContext类型的定义，请参见InferenceContext。

约束说明

无

调用示例

ge::graphStatus InferShapeForXXX(CtInferShapeContext *context) {
  const auto &read_inference_context = ct_context->GetInferenceContext();
  // ...
}