From fa66cc77ff58f6e4cc6a62cf5c3cff1ab648ed68 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Thu, 3 Aug 2017 15:27:30 +0800 Subject: [PATCH] Mean Op Merge Develop --- paddle/operators/mean_op.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/operators/mean_op.h b/paddle/operators/mean_op.h index a89cb422f9..e712dee6a7 100644 --- 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::Flatten(*IG).device(*(context.GetEigenDevice())) = + EigenVector::Flatten(*IG).device(context.GetEigenDevice()) = EigenScalar::From(*OG) / ig_size; } }; -- GitLab