fix (#134)

eric-haibin-lin · Jul 30, 2017 · 0a45136 · 0a45136
1 parent 9321635
commit 0a45136
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/core/symbolic.cc b/src/core/symbolic.cc
@@ -281,13 +281,13 @@ void Symbol::Compose(const array_view<const Symbol*>& args,
     CHECK_EQ(kv.second->outputs.size(), 1U)
         << "Keyword Argument " << kv.first << " is a tuple, single value is required";
   }
+  // assign new name
+  if (!name.empty()) outputs[0].node->attrs.name = name;
 
   // Atomic functor composition.
   if (IsAtomic(outputs)) {
     Node* n = outputs[0].node.get();
     uint32_t n_req = n->num_inputs();
-    // assign new name
-    if (!name.empty()) n->attrs.name = name;
 
     if (n_req != kVarg) {
       n->inputs.resize(n_req);