diff --git a/nnvm/src/core/symbolic.cc b/nnvm/src/core/symbolic.cc index 535dfa4f2127..d3a601d41424 100644 --- a/nnvm/src/core/symbolic.cc +++ b/nnvm/src/core/symbolic.cc @@ -299,6 +299,8 @@ void Symbol::Compose(const array_view& args, } else { n->inputs[i] = NodeEntry{ 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; } }