Consider version too apache#508 (apache#514)

tqchen · May 29, 2018 · 90c1157 · 90c1157
1 parent a9d2149
commit 90c1157
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);
  }