Commit 0b88df6d by Wuwei Lin Committed by Jared Roesch

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

parent 05a5c170
...@@ -101,6 +101,7 @@ ...@@ -101,6 +101,7 @@
namespace tvm { namespace tvm {
namespace relay { namespace relay {
namespace partial_eval {
using namespace runtime; using namespace runtime;
...@@ -1087,12 +1088,14 @@ Expr PostProcess(const Expr& e) { ...@@ -1087,12 +1088,14 @@ Expr PostProcess(const Expr& e) {
return StripWithFuncId(DeDup(Remap(e))); return StripWithFuncId(DeDup(Remap(e)));
} }
} // namespace partial_eval
Expr PartialEval(const Expr& e, const Module& m) { Expr PartialEval(const Expr& e, const Module& m) {
return TransformF([&](const Expr& e) { return TransformF([&](const Expr& e) {
return LetList::With([&](LetList* ll) { return LetList::With([&](LetList* ll) {
PartialEvaluator pe(FreeVars(e), m); relay::partial_eval::PartialEvaluator pe(FreeVars(e), m);
pe.InitializeFuncId(e); pe.InitializeFuncId(e);
return PostProcess(pe.VisitExpr(e, ll)->dynamic); return relay::partial_eval::PostProcess(pe.VisitExpr(e, ll)->dynamic);
}); });
}, e); }, e);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment