onnxruntime

Add shape inference support for a few training ops

#5660

Merged