Commit Graph

6 Commits (a5fc291fe5861417f0648b9cbd061ea0212ce2d5)

Author SHA1 Message Date
danleifeng 9171f73714 fix fp16 grid_size for size=1; test=develop (#20812)
5 years ago
danleifeng 425279a57b Improve elementwise operators performance in same dimensions. (#19763)
5 years ago
lvmengsi 977e9fcb27
support elementwise_sub double backward (#17476)
6 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
6 years ago
Yu Yang 7604b1ad51 Fix Eigen macro when using GPU
6 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
6 years ago