[caffe2][glow] Share info about current batch_size (#58902)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58902
Pull Request resolved: https://github.com/pytorch/glow/pull/5681
Reviewed By: ChunliF
Differential Revision: D28665162
fbshipit-source-id: 39e173a24ee247bc6fee44009798c74dddb27648
