[XLA:CollectiveTransformationReorderer] Move all-reduce from the bottom of the reshape chain to the top as long as reshape has a single user and all-reduce is not layout constrained.
[XLA:CollectiveTransformationReorderer] Move all-reduce from the bottom of the reshape chain to the top as long as reshape has a single user and all-reduce is not layout constrained.
