GetInferenceContextPtr

函数功能

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

函数原型

const ge::InferenceContext *GetInferenceContextPtr() const

参数说明

无。

返回值说明

返回InferenceContext指针。

约束说明

无。

调用示例

ge::graphStatus InferShapeForResourceOp(InferShapeContext *context) {
  auto inference_ctx = context->GetInferenceContextPtr();        // 资源类算子获取InferenceContext
  if (inference_ctx == nullptr) {
    // 防御式编程，不应该出现的场景，打印错误并返回失败
    return ge::GRAPH_FAILED;
  }
  // ...
}