Commit c5a34225 by Tianqi Chen

Enable copy over behavior of attributes of automatic created vars (#34)

parent a6f6a0e0
...@@ -299,6 +299,8 @@ void Symbol::Compose(const array_view<const Symbol*>& args, ...@@ -299,6 +299,8 @@ void Symbol::Compose(const array_view<const Symbol*>& args,
} else { } else {
n->inputs[i] = NodeEntry{ n->inputs[i] = NodeEntry{
CreateVariableNode(DefaultVarName(name, arg_names[i])), 0, 0}; CreateVariableNode(DefaultVarName(name, arg_names[i])), 0, 0};
// copy attribute of parent over automatically created variables
n->inputs[i].node->attrs.dict = n->attrs.dict;
} }
} }
......
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