Consider version too #508 (#514)

srkreddy1238 · tqchen · commit 42f971d1cc06 · 2018-05-29T08:47:00.000-07:00
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);
       }

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,9 @@ nnvm::Graph PrecomputePrune(nnvm::Graph src) {`
`34`	`34`	`}`
`35`	`35`	`nnvm::NodePtr var = nnvm::Node::Create();`
`36`	`36`	`var->attrs.name = e.node->attrs.name;`
	`37`	`+ if (e.version) {`
	`38`	`+ var->attrs.name += "_" + std::to_string(e.version);`
	`39`	`+ }`
`37`	`40`	`if (e.node->num_outputs() != 1) {`
`38`	`41`	`var->attrs.name += "_output" + std::to_string(e.index);`
`39`	`42`	`}`