Simplify lambda syntax in SegmentReduce.cpp (#61416)
Summary:
Fixes Windows build by dismantling a combination nested lambdas+preprocessor magic into explicit templates
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61416
Reviewed By: pbelevich
Differential Revision: D29616449
Pulled By: malfet
fbshipit-source-id: 687ef73b8b37bc272f82d44fc690448e403e3a0c
