提交 d8830802 编写于 作者: Y Yiqun Liu 提交者: GitHub

Merge pull request #4756 from guoshengCS/fix-InferShapeContext-decayed_adagrad_op

Fix InferShapeContext in decayed_adagrad_op
...@@ -22,7 +22,7 @@ class DecayedAdagradOp : public framework::OperatorWithKernel { ...@@ -22,7 +22,7 @@ class DecayedAdagradOp : public framework::OperatorWithKernel {
using framework::OperatorWithKernel::OperatorWithKernel; using framework::OperatorWithKernel::OperatorWithKernel;
protected: protected:
void InferShape(framework::InferShapeContextBase *ctx) const override { void InferShape(framework::InferShapeContext *ctx) const override {
PADDLE_ENFORCE(ctx->HasInput("Param"), PADDLE_ENFORCE(ctx->HasInput("Param"),
"Input(Param) of DecayedAdagradOp should not be null."); "Input(Param) of DecayedAdagradOp should not be null.");
PADDLE_ENFORCE(ctx->HasInput("Grad"), PADDLE_ENFORCE(ctx->HasInput("Grad"),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册