提交 98a0f14e 编写于 作者: 李寅

Merge branch 'fix_fc' into 'master'

fix dataformat transpose for input node

See merge request !1044
......@@ -198,7 +198,7 @@ class Transformer(base_converter.ConverterInterface):
op.output.extend([input_node.name])
output_shape = op.output_shape.add()
output_shape.dims.extend(input_node.shape)
if input_node in self._consumers:
if input_node.name in self._consumers:
if ConverterUtil.data_format(
self._consumers[input_node.name][0]) \
== DataFormat.NCHW:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册