Commit Graph

8 Commits (01ad8d2e069bdd149d3bcbb07e73ab784374a501)

Author SHA1 Message Date
danleifeng 425279a57b Improve elementwise operators performance in same dimensions. (#19763)
6 years ago
Leo Chen 982e61f5ff Update elementwise double grad to save gpu memory (#19509)
6 years ago
lvmengsi 10b23a72c1 Double backward elementwise div (#17416)
6 years ago
Wu Yi 856f0da0fe
Fp16 training (#14992)
6 years ago
typhoonzero da87f7a698 Revert "[Feature] Fp16 training for resnet50 (#14850)"
7 years ago
Wu Yi 3d750f9c5a
[Feature] Fp16 training for resnet50 (#14850)
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