From 521514da361059dcad0b918f27081fdedccf6720 Mon Sep 17 00:00:00 2001 From: QI JUN Date: Tue, 17 Oct 2017 20:01:23 -0700 Subject: [PATCH] fix conflict (#4883) --- paddle/operators/momentum_op.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/operators/momentum_op.cc b/paddle/operators/momentum_op.cc index efa0b59992..9be4d15a43 100644 --- a/paddle/operators/momentum_op.cc +++ b/paddle/operators/momentum_op.cc @@ -22,7 +22,7 @@ class MomentumOp : 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 Momentum should not be null."); PADDLE_ENFORCE(ctx->HasInput("Grad"), -- GitLab