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

tqchen · tqchen · commit 200973ad6d08 · 2018-05-29T08:47:00.000-07:00
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;
           }
         }
 

Original file line number	Diff line number	Diff line change
`@@ -299,6 +299,8 @@ void Symbol::Compose(const array_view<const Symbol*>& args,`
`299`	`299`	`} else {`
`300`	`300`	`n->inputs[i] = NodeEntry{`
`301`	`301`	`CreateVariableNode(DefaultVarName(name, arg_names[i])), 0, 0};`
	`302`	`+ // copy attribute of parent over automatically created variables`
	`303`	`+ n->inputs[i].node->attrs.dict = n->attrs.dict;`
`302`	`304`	`}`
`303`	`305`	`}`
`304`	`306`