Skip to content

Mace
Mace

慢慢CG / Mace
与 Fork 源项目一致

Fork自 Xiaomi / Mace

通知 1
Star 0
Fork 0
Mace
Mace
提交 d8d6ceab 编写于 作者: L liutuo
浏览文件
操作

fix b2i bug for winograd inverse transform

上级 50666a81
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
mace/python/tools/converter_tool/transformer.py
浏览文件 @ d8d6ceab
...@@ -1174,7 +1174,7 @@ class Transformer(base_converter.ConverterInterface): ...@@ -1174,7 +1174,7 @@ class Transformer(base_converter.ConverterInterface):
self.buffer_to_image(op, 0, OpenCLBufferType.WINOGRAD_FILTER) self.buffer_to_image(op, 0, OpenCLBufferType.WINOGRAD_FILTER)
elif op.type == MaceOp.WinogradInverseTransform.name \ elif op.type == MaceOp.WinogradInverseTransform.name \
and len(op.input) >= 3: 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: elif op.type == MaceOp.FullyConnected.name:
self.buffer_to_image(op, 1, OpenCLBufferType.WEIGHT_WIDTH) self.buffer_to_image(op, 1, OpenCLBufferType.WEIGHT_WIDTH)
if len(op.input) >= 3: if len(op.input) >= 3:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或