[Relay] Fix name conflict in PartialEval (#3402)
authorWuwei Lin <wuwelin@amazon.com>
Thu, 20 Jun 2019 07:25:37 +0000 (15:25 +0800)
committerJared Roesch <roeschinc@gmail.com>
Thu, 20 Jun 2019 07:25:37 +0000 (00:25 -0700)
src/relay/pass/partial_eval.cc

index 07ec1b0..f1ca573 100644 (file)
 
 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);
 }