Commit 8d39aaba by Cyril Lashkevich Committed by Tianqi Chen
parent ddbbacc3
...@@ -269,6 +269,11 @@ def PaddingLayerParams(op, insym, symtab): ...@@ -269,6 +269,11 @@ def PaddingLayerParams(op, insym, symtab):
raise NotImplementedError("Only constant padding is supported now.") raise NotImplementedError("Only constant padding is supported now.")
return insym return insym
def PermuteLayerParams(op, insym, symtab):
axes = tuple(op.axis)
return _sym.transpose(insym, axes=axes)
_convert_map = { _convert_map = {
'NeuralNetworkMeanImage': NeuralNetworkMeanImage, 'NeuralNetworkMeanImage': NeuralNetworkMeanImage,
'NeuralNetworkImageScaler': NeuralNetworkImageScaler, 'NeuralNetworkImageScaler': NeuralNetworkImageScaler,
...@@ -283,6 +288,7 @@ _convert_map = { ...@@ -283,6 +288,7 @@ _convert_map = {
'FlattenLayerParams':FlattenLayerParams, 'FlattenLayerParams':FlattenLayerParams,
'ConcatLayerParams':ConcatLayerParams, 'ConcatLayerParams':ConcatLayerParams,
'PaddingLayerParams':PaddingLayerParams, 'PaddingLayerParams':PaddingLayerParams,
'PermuteLayerParams':PermuteLayerParams,
} }
def coreml_op_to_nnvm(op, inname, outname, symtab): def coreml_op_to_nnvm(op, inname, outname, 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