diff --git a/tools/python/transform/transformer.py b/tools/python/transform/transformer.py index 881c5357b4e21545d9a0a759eae839a5a66b05a4..4a27465b409074b2c4b003a44020bc92d0679e91 100644 --- a/tools/python/transform/transformer.py +++ b/tools/python/transform/transformer.py @@ -1115,7 +1115,9 @@ class Transformer(base_converter.ConverterInterface): if (op.type == MaceOp.Conv2D.name or op.type == MaceOp.Deconv2D.name or (op.type == MaceOp.DepthwiseConv2d.name and - self._option.device == DeviceType.APU.value)) and\ + self._option.device == DeviceType.APU.value) or + (op.type == MaceOp.FullyConnected.name and + len(self._consts[op.input[1]].dims) == 4)) and \ op.input[1] not in transposed_filter: filter = self._consts[op.input[1]] filter_data = np.array(filter.float_data).reshape(