Consider version too #508 (#514)

apache · May 29, 2018 · 42f971d · 42f971d
1 parent 9f0def1
commit 42f971d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nnvm/src/compiler/precompute_prune.cc b/nnvm/src/compiler/precompute_prune.cc
@@ -34,6 +34,9 @@ nnvm::Graph PrecomputePrune(nnvm::Graph src) {
       }
       nnvm::NodePtr var = nnvm::Node::Create();
       var->attrs.name = e.node->attrs.name;
+      if (e.version) {
+          var->attrs.name += "_" + std::to_string(e.version);
+      }
       if (e.node->num_outputs() != 1) {
         var->attrs.name += "_output" + std::to_string(e.index);
       }