Commit Graph

4 Commits (2a9d74f67c39d46986868faa0700a705e7d57a24)

Author SHA1 Message Date
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
7 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
7 years ago