From 41ced8e2af40358e3052cd2933cf9694ef05f095 Mon Sep 17 00:00:00 2001
From: fengjiayi <fengjiayi@baidu.com>
Date: Wed, 6 Jun 2018 14:07:29 +0800
Subject: [PATCH] Refine code

---
 paddle/fluid/operators/reverse_op.cc | 7 -------
 paddle/fluid/operators/reverse_op.cu | 7 -------
 paddle/fluid/operators/reverse_op.h  | 4 +++-
 3 files changed, 3 insertions(+), 15 deletions(-)

diff --git a/paddle/fluid/operators/reverse_op.cc b/paddle/fluid/operators/reverse_op.cc
index 64c4c37bb..a20f7d231 100644
--- a/paddle/fluid/operators/reverse_op.cc
+++ b/paddle/fluid/operators/reverse_op.cc
@@ -105,10 +105,3 @@ REGISTER_OP_CPU_KERNEL(
     ops::ReverseKernel<paddle::platform::CPUDeviceContext, bool>,
     ops::ReverseKernel<paddle::platform::CPUDeviceContext, float>,
     ops::ReverseKernel<paddle::platform::CPUDeviceContext, double>)
-REGISTER_OP_CPU_KERNEL(
-    reverse_grad, ops::ReverseKernel<paddle::platform::CPUDeviceContext, int>,
-    ops::ReverseKernel<paddle::platform::CPUDeviceContext, uint8_t>,
-    ops::ReverseKernel<paddle::platform::CPUDeviceContext, int64_t>,
-    ops::ReverseKernel<paddle::platform::CPUDeviceContext, bool>,
-    ops::ReverseKernel<paddle::platform::CPUDeviceContext, float>,
-    ops::ReverseKernel<paddle::platform::CPUDeviceContext, double>)
diff --git a/paddle/fluid/operators/reverse_op.cu b/paddle/fluid/operators/reverse_op.cu
index 7a750c75b..635c41529 100644
--- a/paddle/fluid/operators/reverse_op.cu
+++ b/paddle/fluid/operators/reverse_op.cu
@@ -22,10 +22,3 @@ REGISTER_OP_CUDA_KERNEL(
     ops::ReverseKernel<paddle::platform::CUDADeviceContext, bool>,
     ops::ReverseKernel<paddle::platform::CUDADeviceContext, float>,
     ops::ReverseKernel<paddle::platform::CUDADeviceContext, double>)
-REGISTER_OP_CUDA_KERNEL(
-    reverse_grad, ops::ReverseKernel<paddle::platform::CUDADeviceContext, int>,
-    ops::ReverseKernel<paddle::platform::CUDADeviceContext, uint8_t>,
-    ops::ReverseKernel<paddle::platform::CUDADeviceContext, int64_t>,
-    ops::ReverseKernel<paddle::platform::CUDADeviceContext, bool>,
-    ops::ReverseKernel<paddle::platform::CUDADeviceContext, float>,
-    ops::ReverseKernel<paddle::platform::CUDADeviceContext, double>)
diff --git a/paddle/fluid/operators/reverse_op.h b/paddle/fluid/operators/reverse_op.h
index 2af4268df..d3c54b731 100644
--- a/paddle/fluid/operators/reverse_op.h
+++ b/paddle/fluid/operators/reverse_op.h
@@ -24,7 +24,9 @@ struct ReverseFunctor {
   void operator()(const DeviceContext& context, const framework::LoDTensor& in,
                   framework::LoDTensor* out, const std::vector<int>& axis) {
     Eigen::array<bool, Rank> reverse_axis;
-    reverse_axis.fill(false);
+    for (auto& i : reverse_axis) {
+      i = false;
+    }
     for (int a : axis) {
       reverse_axis[a] = true;
     }
-- 
GitLab