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