• Amy Zhuang's avatar
    Use Eigen kernel for ScatterAdd. (#3002) · ac17d797
    Amy Zhuang authored
    *  Use Eigen kernel for ScatterAdd.
    
    * Emit ScatterAdd Eigen kernel for CODEGEN.
    
    Add comments.
    
    Address PR feedback.
    
    * Add more unit tests.
    
    Fix style error.
    
    Add ScatterAdd to AssignOpMap.
    
    * Combine non-scalar cases together.
    
    * Address PR feedback.
    
    * Fix a bug.
    
    * Use reshape to make the shapes of two slices match.
    
    * Rename variables.
    
    Fix bugs.
    
    Use helper function.
    
    Add one unit test.
    
    * Add reshape back.
    ac17d797
Name
Last commit
Last update
.ci Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
.clang-format Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.travis.yml Loading commit data...
ABOUT.md Loading commit data...
CMakeLists.txt Loading commit data...
CODEOWNERS Loading commit data...
CONTRIB.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...
ecosystem-overview.md Loading commit data...