From 65aac6637f1d124f525930d2f8f77f6f17af6db9 Mon Sep 17 00:00:00 2001 From: Bin Li Date: Fri, 18 Jan 2019 10:48:29 +0800 Subject: [PATCH] Support align_corners for Hexagon ResizeBilinear --- mace/python/tools/converter_tool/hexagon_converter.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/mace/python/tools/converter_tool/hexagon_converter.py b/mace/python/tools/converter_tool/hexagon_converter.py index 42154b99..05618e9c 100644 --- a/mace/python/tools/converter_tool/hexagon_converter.py +++ b/mace/python/tools/converter_tool/hexagon_converter.py @@ -232,6 +232,14 @@ class HexagonConverter(base_converter.ConverterInterface): newdim_tensor.int32_data.extend(newdim_arg.ints) op.input.extend([newdim_tensor.name]) self.add_min_max_const_node(op, op.input[0]) + align_corners_arg = ConverterUtil.get_arg( + op, MaceKeyword.mace_align_corners_str) + align_corners_tensor = self._model.tensors.add() + align_corners_tensor.name = op.name + '/align_corners:0' + align_corners_tensor.data_type = mace_pb2.DT_INT32 + align_corners_tensor.dims.extend([1]) + align_corners_tensor.int32_data.extend([align_corners_arg.i]) + op.input.extend([align_corners_tensor.name]) elif op.type == MaceOp.Concat.name: inputs = copy.deepcopy(op.input) for ipt in inputs: -- GitLab