We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ab65226 commit 4f67977Copy full SHA for 4f67977
paddle/fluid/operators/pull_gpups_sparse_op.cc
@@ -117,7 +117,8 @@ class PushGpuPSSparseOpMaker : public framework::SingleGradOpMaker<T> {
117
op->SetType("push_gpups_sparse");
118
op->SetInput("Ids", this->Input("Ids"));
119
op->SetInput(framework::GradVarName("Out"), this->OutputGrad("Out"));
120
- op->SetOutput(framework::GradVarName("Ids"), this->InputGrad("Ids", false));
+ op->SetOutput(framework::GradVarName("Ids"),
121
+ this->InputGrad("Ids", /* drop_empty_grad */ false));
122
op->SetAttrMap(this->Attrs());
123
}
124
};
0 commit comments