Fix bug introduced in batch_norm

apache · Apr 17, 2019 · 9c74556 · 9c74556
1 parent 20c4a1a
commit 9c74556
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/operator/nn/batch_norm.cc b/src/operator/nn/batch_norm.cc
@@ -485,9 +485,8 @@ std::vector<nnvm::NodeEntry> BatchNormGrad(const nnvm::NodePtr& n,
                                            const std::vector<nnvm::NodeEntry>& ograds) {
   std::vector<nnvm::NodeEntry> out_data(n->num_outputs());
   out_data.reserve(n->num_outputs());
-  for (size_t i = 0; i < out_data.size(); ++i) {
+  for (size_t i = 0; i < n->num_outputs(); ++i)
     out_data.emplace_back(n, i, 0);
-  }
   std::vector<nnvm::NodeEntry> heads;
   heads.reserve(8);
   heads.emplace_back(ograds[0]);