From 9c67c31b42875819e5a6f641ee1497368effffe4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8F=B6=E5=89=91=E6=AD=A6?= Date: Fri, 28 Jun 2019 13:47:15 +0800 Subject: [PATCH] register uint8 reshape for cpu remove identity in-out reshape op in converter --- mace/python/tools/converter_tool/transformer.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py index 1a9b7993..f4d461ed 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -356,6 +356,13 @@ class Transformer(base_converter.ConverterInterface): self.safe_remove_node(op, self._producer.get(op.input[0], None)) return True + if op.type == 'Reshape' and \ + op.output_shape[0].dims == \ + self.get_tensor_shape(op.input[0]): + print("Remove useless reshape: %s(%s)" % (op.name, op.type)) + self.safe_remove_node(op, + self._producer.get(op.input[0], None)) + return True return False -- GitLab