onnxruntime

fed52053 - Refine a bit (on device training) (#15803)

Refine a bit (on device training) (#15803) ### Few minor refinements: - Simplify ParameterOptimizerState a bit - Use inlined containers - Remove GetStateDict APIs] - Re-enable cuda test for lr scheduler