fix bug

28476676 · fengjiayi · 5d2b3fb0 · 28476676
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/operators/fill_zeros_like_op.h paddle/operators/fill_zeros_like_op.h +2 -1

未找到文件。
--- a/paddle/operators/fill_zeros_like_op.h
+++ b/paddle/operators/fill_zeros_like_op.h
@@ -24,7 +24,8 @@ class FillZerosLikeKernel : public framework::OpKernel {
  void Compute(const framework::ExecutionContext& context) const override {
    auto* output = context.Output<framework::Tensor>(0);
    output->mutable_data<T>(context.GetPlace());
-    framework::EigenVector<T>::Flatten(*output).setZero();
+    auto t = framework::EigenVector<T>::Flatten(*output);
+    t.device(context.GetEigenDevice<Place>()) = t.constant(T(0));
  }
 };