提交 cc2908e7 编写于 作者: L Liangliang He

Update conv2d filter from HWIO to HWOI

上级 e71a13c4
......@@ -315,6 +315,7 @@ class TFConverter(object):
op_def.type = 'DepthwiseConv2d'
else:
op_def.type = op.type
self.transpose_filter_tensor[get_input_tensor(op, 1).name] = (0, 1, 3, 2)
if self.device == 'gpu':
op_def.input.extend([op.inputs[0].name])
buffer_type = "DW_CONV2D_FILTER" if op_def.type == 'DepthwiseConv2d' else "CONV2D_FILTER"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册