Fixed attribute parameter (#184)

eric-haibin-lin · Oct 19, 2017 · 69bea12 · 69bea12
1 parent 8d79cfd
commit 69bea12
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/nnvm/frontend/coreml.py b/python/nnvm/frontend/coreml.py
@@ -246,7 +246,7 @@ def ConcatLayerParams(op, insyms, symtab):
         insyms = [insyms]
     if op.sequenceConcat:
         raise NotImplementedError("Sequence Concat not supported")
-    ret = _sym.concatenate(*insyms, dim=1)
+    ret = _sym.concatenate(*insyms, axis=1)
     return ret
 
 def FlattenLayerParams(op, insym, symtab):