Commit Graph

6 Commits (f56967c48388fe4a365c3b63be09694008b46232)

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