Testing of masked reductions on mixed layout inputs. (#72398)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/72398
The design of this feature is discussed in https://github.com/pytorch/pytorch/pull/71239#discussion_r787292751
Test Plan: Imported from OSS
Reviewed By: george-qi
Differential Revision: D34408881
Pulled By: cpuhrsch
fbshipit-source-id: a362d4220957ea38b7e442df4ecf260ffe682eab
(cherry picked from commit 7fb3611130c08f1aa6ea708ca838708c13b0f01c)
