提交 c20544bb 编写于 作者: 刘琦 提交者: 赵奇可

Merge branch 'fix_b2i_for_winograd' into 'master'

fix b2i bug for winograd inverse transform

See merge request !776
......@@ -1327,7 +1327,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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册