Skip to content

T
tic
Commit 90c1157b by Siva Committed by Tianqi Chen
Options

Consider version too #508 (#514)

parent a9d21497
Showing with 3 additions and 0 deletions
nnvm/src/compiler/precompute_prune.cc
...@@ -34,6 +34,9 @@ nnvm::Graph PrecomputePrune(nnvm::Graph src) { ...@@ -34,6 +34,9 @@ nnvm::Graph PrecomputePrune(nnvm::Graph src) {
} }
nnvm::NodePtr var = nnvm::Node::Create(); nnvm::NodePtr var = nnvm::Node::Create();
var->attrs.name = e.node->attrs.name; var->attrs.name = e.node->attrs.name;
if (e.version) {
var->attrs.name += "_" + std::to_string(e.version);
}
if (e.node->num_outputs() != 1) { if (e.node->num_outputs() != 1) {
var->attrs.name += "_output" + std::to_string(e.index); var->attrs.name += "_output" + std::to_string(e.index);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment