From 2530c7eef635eecc59da0ac1a4e7a87c986d42e1 Mon Sep 17 00:00:00 2001 From: Bin Li Date: Sat, 9 May 2020 17:52:07 +0800 Subject: [PATCH] Fix ONNX padding and add transpose half --- mace/ops/transpose.cc | 2 ++ tools/python/transform/onnx_converter.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/mace/ops/transpose.cc b/mace/ops/transpose.cc index 3a5eb68b..2d15d053 100644 --- a/mace/ops/transpose.cc +++ b/mace/ops/transpose.cc @@ -65,6 +65,8 @@ class TransposeOp : public Operation { void RegisterTranspose(OpRegistry *op_registry) { MACE_REGISTER_OP(op_registry, "Transpose", TransposeOp, DeviceType::CPU, float); + MACE_REGISTER_OP(op_registry, "Transpose", TransposeOp, + DeviceType::CPU, half); MACE_REGISTER_BF16_OP(op_registry, "Transpose", TransposeOp, DeviceType::CPU); } diff --git a/tools/python/transform/onnx_converter.py b/tools/python/transform/onnx_converter.py index 00e0aca8..62b380c3 100644 --- a/tools/python/transform/onnx_converter.py +++ b/tools/python/transform/onnx_converter.py @@ -1236,6 +1236,8 @@ class OnnxConverter(base_converter.ConverterInterface): paddings_arg = op.arg.add() paddings_arg.name = MaceKeyword.mace_paddings_str paddings_value = node.attrs['pads'] + paddings_value = np.asarray(paddings_value).reshape( + (2, -1)).transpose().reshape(-1).tolist() paddings_arg.ints.extend(paddings_value) if 'value' in node.attrs: constant_value_arg = op.arg.add() -- GitLab