Commit Graph

7 Commits (cced930b61ba246dffec68bbe09bd9e22a142d64)

Author SHA1 Message Date
chentianyu03 8f45d14263
add complex64 and complex128 type; add +-*/@ and slice opreator for c… (#29199)
5 years ago
danleifeng 9171f73714 fix fp16 grid_size for size=1; test=develop (#20812)
7 years ago
danleifeng 425279a57b Improve elementwise operators performance in same dimensions. (#19763)
7 years ago
lvmengsi 977e9fcb27
support elementwise_sub double backward (#17476)
7 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)
8 years ago