获取当前算子传递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; } // ... }