onnxruntime

5129b4d5 - batch size tests (#5508)

batch size tests (#5508) * batch size tests Co-authored-by: liqun <liqun@OrtTrainingDev4.af05slrtruoetgaxwwjv5nsq5e.px.internal.cloudapp.net>