Commit 0b88df6d by Wuwei Lin Committed by Jared Roesch

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

parent 05a5c170
......@@ -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);
}
......
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