onnxruntime

Add an option to save the training graph after optimization

#6410

Merged