pytorch
921d2a16 - SparseAdagrad/RowWiseSparseAdagrad mean fusion on CPU & GPU and dedup version for RowWiseSparse mean fusion on GPU

Commit
4 years ago
SparseAdagrad/RowWiseSparseAdagrad mean fusion on CPU & GPU and dedup version for RowWiseSparse mean fusion on GPU Summary: 1. Support SparseAdagradFusedWithSparseLengthsMeanGradient and RowWiseSparseAdagradFusedWithSparseLengthsMeanGradient on CPU and GPU 2. Add the dedup implementation of fused RowWiseAdagrad op on GPUs for mean pooling Reviewed By: xianjiec Differential Revision: D22165603 fbshipit-source-id: 743fa55ed5893c34bc6406ddfbbbb347b88091d1
Author
Yan Xie
Parents
Loading