diff --git a/tools/python/transform/transformer.py b/tools/python/transform/transformer.py index d374ec46cc28f2bf924c225391fcc6981491807d..778dcb1fd015fb40b348bbb7b11d998bd11e947f 100644 --- a/tools/python/transform/transformer.py +++ b/tools/python/transform/transformer.py @@ -1013,7 +1013,8 @@ class Transformer(base_converter.ConverterInterface): zero_padding = False if height == filter_height and width == filter_width \ - and zero_padding: + and zero_padding \ + and len(self._consumers[op.input[1]]) == 1: print("transform global conv to fc %s(%s)" % (op.name, op.type)) op.type = MaceOp.FullyConnected.name