Mean Op Merge Develop

8 years ago · fa66cc77ff
parent 858b97cbea
commit fa66cc77ff
1 changed files with 1 additions and 1 deletions
--- a/paddle/operators/mean_op.h
+++ b/paddle/operators/mean_op.h
@ -47,7 +47,7 @@ public:

    T ig_size = (T)framework::product(IG->dims());

-    EigenVector<T>::Flatten(*IG).device(*(context.GetEigenDevice<Place>())) =
+    EigenVector<T>::Flatten(*IG).device(context.GetEigenDevice<Place>()) =
        EigenScalar<T>::From(*OG) / ig_size;
  }
 };