From d0d3129f914c1954e5fca9bdfa3653e4bbf2a3ff Mon Sep 17 00:00:00 2001 From: guosheng Date: Thu, 12 Oct 2017 19:16:46 +0800 Subject: [PATCH] Fix InferShapeContext in decayed_adagrad_op --- paddle/operators/decayed_adagrad_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/operators/decayed_adagrad_op.cc b/paddle/operators/decayed_adagrad_op.cc index ca5141dabc2..7f583f18c8c 100644 --- a/paddle/operators/decayed_adagrad_op.cc +++ b/paddle/operators/decayed_adagrad_op.cc @@ -22,7 +22,7 @@ class DecayedAdagradOp : public framework::OperatorWithKernel { using framework::OperatorWithKernel::OperatorWithKernel; protected: - void InferShape(framework::InferShapeContextBase *ctx) const override { + void InferShape(framework::InferShapeContext *ctx) const override { PADDLE_ENFORCE(ctx->HasInput("Param"), "Input(Param) of DecayedAdagradOp should not be null."); PADDLE_ENFORCE(ctx->HasInput("Grad"), -- GitLab