pytorch

1ec30a66 - Add offsets-based reduction to segment_reduce (CPU, CUDA)

Add offsets-based reduction to segment_reduce (CPU, CUDA) Pull Request resolved: https://github.com/pytorch/pytorch/pull/78907 Approved by: https://github.com/cpuhrsch