From 5fb9bdc892c921aab41ff79b4296546b2b071b97 Mon Sep 17 00:00:00 2001 From: heqiaozhi Date: Thu, 11 Apr 2019 19:58:18 +0800 Subject: [PATCH] add X to grad test=develop --- paddle/fluid/operators/cvm_op.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/fluid/operators/cvm_op.cc b/paddle/fluid/operators/cvm_op.cc index eb940c509..53ed86ade 100644 --- a/paddle/fluid/operators/cvm_op.cc +++ b/paddle/fluid/operators/cvm_op.cc @@ -131,6 +131,7 @@ class CVMGradOpDescMaker : public framework::SingleGradOpDescMaker { std::unique_ptr Apply() const override { std::unique_ptr op(new framework::OpDesc()); op->SetType("cvm_grad"); + op->SetInput("X", Input("X")); op->SetInput("CVM", Input("CVM")); op->SetInput(framework::GradVarName("Y"), OutputGrad("Y")); op->SetOutput(framework::GradVarName("X"), InputGrad("X")); -- GitLab