fold constant

apache · Jun 1, 2019 · 940eb96 · 940eb96
1 parent 6f83f56
commit 940eb96
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc
@@ -297,6 +297,7 @@ class RelayBuildModule : public runtime::ModuleNode {
     });
     pass_seqs.push_back(transform::EliminateCommonSubexpr(fskip));
     pass_seqs.push_back(transform::CombineParallelConv2D(3));
+    pass_seqs.push_back(transform::FoldConstant());
     pass_seqs.push_back(transform::FoldScaleAxis());
     pass_seqs.push_back(transform::CanonicalizeOps());
 

diff --git a/src/relay/pass/fold_scale_axis.cc b/src/relay/pass/fold_scale_axis.cc
@@ -978,7 +978,7 @@ Pass FoldScaleAxis() {
   // FoldScaleAxis pass contains the following three passes. Therefore, we can
   // register it as a sequential pass.
   Pass pass = Sequential(
-      {FoldConstant(), BackwardFoldScaleAxis(), ForwardFoldScaleAxis()},
+      {BackwardFoldScaleAxis(), ForwardFoldScaleAxis(), FoldConstant()},
       "FoldScaleAxis");
   return pass;
 }