add X to grad

test=develop

add X to grad
test=develop
5fb9bdc8 · heqiaozhi · afa64a5c · 5fb9bdc8
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/cvm_op.cc paddle/fluid/operators/cvm_op.cc +1 -0

未找到文件。
--- 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<framework::OpDesc> Apply() const override {
    std::unique_ptr<framework::OpDesc> 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"));