Moderate the dim type after LengthsRangeFill (#17096)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17096
LengthsRangeFill will take a batch size of lengths input and expand it into sequence. Later op should follow this type until it hits another batch type moderating op, e.g. SparseLengthsSum.
Reviewed By: ipiszy
Differential Revision: D14079422
fbshipit-source-id: 1a26925d502c32875ea95c160268bf6a256cc955