diff --git a/tools/python/transform/transformer.py b/tools/python/transform/transformer.py index bbc94f5cee0ba9dd609f25e224d8740b2adb0d0d..19a2921c0aeeb4f25b836a067f56db9de99817b0 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(