[Relay] Fix name conflict in PartialEval (#3402)

apache · Jun 20, 2019 · 0b88df6 · 0b88df6
1 parent 05a5c17
commit 0b88df6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/relay/pass/partial_eval.cc b/src/relay/pass/partial_eval.cc
@@ -101,6 +101,7 @@
 
 namespace tvm {
 namespace relay {
+namespace partial_eval {
 
 using namespace runtime;
 
@@ -1087,12 +1088,14 @@ Expr PostProcess(const Expr& e) {
   return StripWithFuncId(DeDup(Remap(e)));
 }
 
+}  // namespace partial_eval
+
 Expr PartialEval(const Expr& e, const Module& m) {
   return TransformF([&](const Expr& e) {
       return LetList::With([&](LetList* ll) {
-          PartialEvaluator pe(FreeVars(e), m);
+          relay::partial_eval::PartialEvaluator pe(FreeVars(e), m);
           pe.InitializeFuncId(e);
-          return PostProcess(pe.VisitExpr(e, ll)->dynamic);
+          return relay::partial_eval::PostProcess(pe.VisitExpr(e, ll)->dynamic);
         });
     }, e);
 }