提交 45cd1f4d 编写于 作者: Y yejianwu

fix dataformat transpose for input node

上级 abbc3d4d
......@@ -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.
先完成此消息的编辑!
想要评论请 注册