fixed map creation logic

apache · Feb 13, 2020 · fdd477b · fdd477b
1 parent 3631bc7
commit fdd477b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/operator/subgraph/build_subgraph.cc b/src/operator/subgraph/build_subgraph.cc
@@ -502,7 +502,7 @@ void FindOutputEntries(nnvm::Graph* g,
             if (node2idx.find(e) == node2idx.end())
               node2idx[e] = node2idx.size();
             uint32_t i = node2idx[e];
-            if (output_map->size() < i)
+            if (output_map->size() > i)
               output_map->at(i).push_back(&e);
             else
               output_map->push_back({&e});
@@ -516,7 +516,7 @@ void FindOutputEntries(nnvm::Graph* g,
           if (node2idx.find(e) == node2idx.end())
             node2idx[e] = node2idx.size();
           uint32_t i = node2idx[e];
-          if (output_map->size() < i)
+          if (output_map->size() > i)
             output_map->at(i).push_back(&e);
           else
             output_map->push_back({&e});
@@ -537,7 +537,7 @@ void FindOutputEntries(nnvm::Graph* g,
         if (node2idx.find(entry) == node2idx.end())
           node2idx[entry] = node2idx.size();
         uint32_t i = node2idx[entry];
-        if (output_map->size() < i)
+        if (output_map->size() > i)
           output_map->at(i).push_back(&entry);
         else
           output_map->push_back({&entry});