gather_op modified

f6bffd4e · zchen0211 · 2a42a73d · f6bffd4e
隐藏空白更改
内联并排

Showing with 4 addition and 2 deletion

paddle/operators/gather_op.cc paddle/operators/gather_op.cc +4 -2

未找到文件。
--- a/paddle/operators/gather_op.cc
+++ b/paddle/operators/gather_op.cc
@@ -51,8 +51,10 @@ Y = X[Index]
 class GatherGradOp : public framework::OperatorWithKernel {
 protected:
  void InferShape(const framework::InferShapeContext &ctx) const override {
-    ctx.Output<Tensor>("X" + framework::kGradVarSuffix)
+    auto X_grad = ctx.Output<Tensor>(framework::GradVarName("X"));
-        ->Resize(ctx.Input<Tensor>("X")->dims());
+    auto X = ctx.Input<Tensor>("X");
+    X_grad->Resize(X->dims());
  }
 };