Commit Graph

5 Commits (569951c418fb3c9f82cbdde9fda3910cc7033bff)

Author SHA1 Message Date
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