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

apache · May 29, 2018 · 200973a · 200973a
1 parent b44568c
commit 200973a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nnvm/src/core/symbolic.cc b/nnvm/src/core/symbolic.cc
@@ -299,6 +299,8 @@ void Symbol::Compose(const array_view<const Symbol*>& 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;
           }
         }