fix bug

8 years ago · 2847667620
parent 5d2b3fb09d
commit 2847667620
1 changed files with 2 additions and 1 deletions
--- 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));
  }
 };