From 4d4b350695235380eba3e07500d11f2a85326570 Mon Sep 17 00:00:00 2001
From: Andrew Liu <andrewlliu@gmail.com>
Date: Sun, 22 Mar 2020 18:15:49 -0700
Subject: [PATCH] review

---
 src/relay/transforms/gradient_cell.cc | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/relay/transforms/gradient_cell.cc b/src/relay/transforms/gradient_cell.cc
index 0a224908fd74..b5504b3ae7ee 100644
--- a/src/relay/transforms/gradient_cell.cc
+++ b/src/relay/transforms/gradient_cell.cc
@@ -37,7 +37,7 @@
  * Note: this pass can only be used with functions where the input/output types are
  * a combination of TupleTypes and TensorTypes
  * 
- * Specify optimize 6 ops:
+ * This pass optimizes 6 ops:
  * - add
  * - multiply
  * - ones
@@ -66,8 +66,6 @@
 namespace tvm {
 namespace relay {
 
-namespace GradientCellPass {  // avoid polluting namespace
-
 /*!
 * \brief Get constructor of GradCell TypeDef with name_hint
 *
@@ -300,10 +298,8 @@ class GradientCellTransform: public ExprMutator, public TypeMutator {
   IRModule module_;
 };
 
-}  // namespace GradientCellPass
-
 Expr GradientCell(const Expr& e, IRModule mod) {
-  return GradientCellPass::GradientCellTransform(mod).transform(e);
+  return GradientCellTransform(mod).transform(e);
 }
 
 namespace transform {