qonnx.util.to_channels_last (module)

qonnx.util.to_channels_last.main()

qonnx.util.to_channels_last.to_channels_last(in_file, *, make_input_channels_last=False, out_file=None)

Execute a set of graph transformations to convert an ONNX file to the channels last data format. The input file have been previously cleaned by the cleanup transformation or commandline tool.

Parameters

• in_file – Filename for the input ONNX model

• make_input_channels_last – Sets if the input of the model should also be converted to the channels last data layout (True) or if a transpose node should be left at the beginning of the graph (False). Defaults to False.

• out_file – If set, filename for the output ONNX model. Set to in_file with _chan_last suffix otherwise.