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

fix dataformat transpose for input node

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