Mean Op Merge Develop

fa66cc77 · Yu Yang · 858b97cb · fa66cc77
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/operators/mean_op.h paddle/operators/mean_op.h +1 -1

未找到文件。
--- 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;
  }
 };