DeepSpeed

f4d18fa2 - fix gates size retrieval logic in _rnn_flops (#3921)

fix gates size retrieval logic in _rnn_flops (#3921) Co-authored-by: Pinstripe Potoroo <pinstripe-potoroo@users.github.com> Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>