Fix a typo in FollowReduceStrategy when creating following strategies for reduce ops. When computing communication and memory resharding costs for an operand of the reduce, we were using the shape of the reduce, instead of the shape of the operand.
Fix a typo in FollowReduceStrategy when creating following strategies for reduce ops. When computing communication and memory resharding costs for an operand of the reduce, we were using the shape of the reduce, instead of the shape of the operand.
