Skip to content

T
tic
Commit ae16a366 by PENGUINLIONG Committed by Tianqi Chen
Options

Fixed attribute parameter (#184)

parent 3833ed2e
Showing with 1 additions and 1 deletions
nnvm/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):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment