Commit Graph

21 Commits (7836f477c6a5c7aac386f512f27399148aa9921d)

Author SHA1 Message Date
gongweibao a7512db2bc
Polish elementwise max min pow document to add more examples. (#19946)
6 years ago
Leo Chen 982e61f5ff Update elementwise double grad to save gpu memory (#19509)
6 years ago
Zeng Jinle 5c8f210ce3
refine inplace inference registry, test=develop (#19032)
6 years ago
Kaipeng Deng bd9bef5a4e
add elementwise_add_grad_grad op (#17366)
6 years ago
Kaipeng Deng 8bae8590ac
add double grad for elementwise_mul op (#17255)
6 years ago
gongweibao 91784f8ec3
Fix code in document. (#17237)
6 years ago
Zeng Jinle ee2028a110
Add use_cuda to inplace pass (#17205)
6 years ago
Zeng Jinle 1c526e1d1a
Fix some grad op desc makers (#16633)
6 years ago
Zeng Jinle c7c6eeb44e
Merge pull request #16409 from sneaxiy/feature/advance_gc
6 years ago
liuwei1031 8d22bc17a4
Memory optimize (#16410)
6 years ago
sneaxiy a93a9eef8f add op registry type
6 years ago
dzhwinter 104d3b4e68 add details. test=develop
6 years ago
dzhwinter 94dd50c33f add details. test=develop
6 years ago
dzhwinter 8f3b252392 squash commits. test=develop
7 years ago
sneaxiy 045dc12766 merge develop
7 years ago
sneaxiy a500dfa579 rewrite ddim
7 years ago
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
7 years ago
chengduo 04539d4c5d
Fix clip.py (#14718)
7 years ago
Michal Gallus 08f63c4d12 MKLDNN elementwise_mul: Lint changes to UT & integration
7 years ago
Michal Gallus ed31936ba1 MKLDNN elementwise_mul: Support NCHW, update UT
7 years ago
Wu Yi a2d9b34417
Refine operator cmake (#14413)
7 years ago