This change adds symbolic shape inference to ORT training which helps static memory planning for model like BART.