pytorch

7360b53f - reland Add offsets-based reduction to segment_reduce (CPU, CUDA)

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