Commit 90c1157b by Siva Committed by Tianqi Chen

Consider version too #508 (#514)

parent a9d21497
......@@ -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);
}
......
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