fix typo on BatchNorm for mirroring (#6541)

* fix bug on BatchNorm for mirroring * keep CuDNNBN
apache · Jun 2, 2017 · 551996a · 551996a
1 parent e41ab86
commit 551996a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/executor/graph_executor.cc b/src/executor/graph_executor.cc
@@ -221,6 +221,7 @@ nnvm::Graph GraphExecutor::InitFullGraph(
     if (type == "FullyConnected") return false;
     if (type == "Concat") return false;
     if (type == "SoftmaxOutput") return false;
+    if (type == "BatchNorm") return false;
     if (type == "CuDNNBatchNorm") return false;
     return true;
   };