pytorch
86f1fedb - Preliminary support of grouped reduction (#1623)

Commit

2 years ago

Preliminary support of grouped reduction (#1623) * Grouped reduction * PR feedback * Consolidate Welford and GroupedReduction rfactor

Author

naoyam

Parents