Commit Graph

18 Commits (a7512db2bc6a5bc6badaaf5b3a32abb42b9463aa)

Author SHA1 Message Date
Zeng Jinle 5c8f210ce3
refine inplace inference registry, test=develop (#19032)
6 years ago
hutuxian 4ff87c049d
remove useless input 'Softmax@GRAD' from softmax_with_cross_entropy op (#17612)
6 years ago
Zeng Jinle 4f8594088d
Enhance inplace/mem-opt pass and enhance softmax_with_cross_entropy op inplace (#17225)
6 years ago
Kaipeng Deng a71d8fdb87
Softmax_cross_entropy op add axis (#16806)
6 years ago
Zeng Jinle ee2028a110
Add use_cuda to inplace pass (#17205)
6 years ago
Zeng Jinle 1202d3fc74
Refine model gpu memory (#16993)
6 years ago
phlrain 5309b081f6 simple code; test=develop
6 years ago
phlrain f7a5a98fdb remove unused code; test=develop
6 years ago
phlrain bbfc82cc42 softmax corss entropy support high rank
6 years ago
sneaxiy 023a3a3d62 fix op grad maker
6 years ago
jerrywgz 5ce48220f1 change default option related to softmax, test=develop
6 years ago
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
6 years ago
sneaxiy 5e5d2223a1 test=develop
6 years ago
Bai Yifan faf8ad2436
Add ignore_index in cross_entropy op (#13217)
7 years ago
Yu Yang 0e78cb69fb Clean OpProtoAndCheckerMaker
7 years ago
qingqing01 24509f4af9 Fix the grammar in copyright. (#8403)
7 years ago
Yi Wang fc374821dd Correct #include path
7 years ago
Yi Wang 90648f336d Move file to fluid/; Edit CMakeLists.txt
7 years ago