From 45cd1f4d42ec3c45e611b89cc288f6294018648c Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Tue, 2 Apr 2019 17:08:17 +0800
Subject: [PATCH] fix dataformat transpose for input node

---
 mace/python/tools/converter_tool/transformer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/python/tools/converter_tool/transformer.py b/mace/python/tools/converter_tool/transformer.py
index 6ce3beba..1083e235 100644
--- a/mace/python/tools/converter_tool/transformer.py
+++ b/mace/python/tools/converter_tool/transformer.py
@@ -198,7 +198,7 @@ class Transformer(base_converter.ConverterInterface):
                     op.output.extend([input_node.name])
                     output_shape = op.output_shape.add()
                     output_shape.dims.extend(input_node.shape)
-                    if input_node in self._consumers:
+                    if input_node.name in self._consumers:
                         if ConverterUtil.data_format(
                                 self._consumers[input_node.name][0]) \
                                 == DataFormat.NCHW:
-- 
GitLab