Unary.cuh
5.15 KB
-
1. add gather function 2. add cross entropy forward computation and backward… · ceb5b1011. add gather function 2. add cross entropy forward computation and backward computation 3. code optimization 4. merge with xiaotong-working branch
xuchen committed