GetInferenceContext

产品支持情况

全量芯片支持。

头文件

#include <graph/ct_infer_shape_range_context.h>

功能说明

获取InferenceContext指针。

函数原型

1

ge::InferenceContext *GetInferenceContext() const

参数说明

无

返回值说明

输出InferenceContext指针。

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

约束说明

无

调用示例

1
2
3
4

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