Commit Graph

16 Commits (9bb6a421e3294dbdcad049f43b1477c5b35f262d)

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