From d8d6ceabcfe287422ba48812420ee85e1991f8fa Mon Sep 17 00:00:00 2001 From: liutuo Date: Fri, 31 Aug 2018 11:14:32 +0800 Subject: [PATCH] fix b2i bug for winograd inverse transform --- 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 dfa92bf3..60cfe88b 100644 --- a/mace/python/tools/converter_tool/transformer.py +++ b/mace/python/tools/converter_tool/transformer.py @@ -1174,7 +1174,7 @@ class Transformer(base_converter.ConverterInterface): self.buffer_to_image(op, 0, OpenCLBufferType.WINOGRAD_FILTER) elif op.type == MaceOp.WinogradInverseTransform.name \ and len(op.input) >= 3: - self.buffer_to_image(op, 1, OpenCLBufferType.ARGUMENT) + self.buffer_to_image(op, 2, OpenCLBufferType.ARGUMENT) elif op.type == MaceOp.FullyConnected.name: self.buffer_to_image(op, 1, OpenCLBufferType.WEIGHT_WIDTH) if len(op.input) >= 3: -- GitLab