Commit Graph

36 Commits (ecd9f330c9ae86414ef31e6808f87a48ce99dac3)

Author SHA1 Message Date
Zeng Jinle 4e1bc6e805
Rewrite inplace pass and fix gc bug (#17126)
6 years ago
Kaipeng Deng 54474637ae
Merge pull request #16057 from heavengate/softmax_axis
6 years ago
liuwei1031 8d22bc17a4
Memory optimize (#16410)
6 years ago
dengkaipeng 90bd038d35 fix format. test=develop
6 years ago
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
6 years ago
dengkaipeng 412b7cbdf1 fix format. test=develop
6 years ago
dengkaipeng 8b88960dce fix doc. test=develop
6 years ago
dengkaipeng 2ddd23dac8 fix format. test=develop
6 years ago
dengkaipeng 217db27337 add mkldnn support. test=develop
6 years ago
dengkaipeng 518325f1e7 add softmax_axis CPU kernel. test=develop
6 years ago
dzhwinter 8f3b252392 squash commits. test=develop
6 years ago
Yu Yang 9bd70a1e04 Change tensor uses proto::VarType::type
6 years ago
Sylwester Fraczek 8a1eeec579 add mkldnn prop_kind phase for inference-only case to pooling and activations (#14278)
6 years ago
chengduo 6c6e638550
Add InferVarType for some op (#14201)
6 years ago
Xin Pan 8f2116d8fa clean up after the changes have been stopped for so long.
6 years ago
fengjiayi b656d97e86
Merge pull request #12485 from JiayiFeng/dev_ops_tensor_support
7 years ago
dzhwinter 595a2c83ae
explicit gradient of elementwise_add/elementwise_sub (#11970)
7 years ago
fengjiayi b314a69523 make softmax supporting tensors
7 years ago
Jacek Czaja 98f3ad3ba1 - MKLDNN Softmax Grad Op
7 years ago
dzhwinter bfa3fd6f15
add inplace attribute to op_proto_maker (#10665)
7 years ago
mozga-intel 3ff9ba0e6b Mkldnn layout (#11040)
7 years ago
Yu Yang 0e78cb69fb Clean OpProtoAndCheckerMaker
7 years ago
dzhwinter f63ff90b03
Fix/fp64 (#10346)
7 years ago
Yang Yang(Tony) e04c43d543
add semicolon to op registry (#10034)
7 years ago
Tao Luo d84cdb7b59
Merge pull request #9911 from tonyyang-svail/unify_op_registry
7 years ago
Jacek Czaja acdf7cbd19 - Added EPS for softmax MKLDNN op
7 years ago
Yang Yang ce7c2e86e5 script to fix all
7 years ago
Kexin Zhao b2a1c9e8b7 Add float16 support to non-cudnn softmax op on GPU (#9686)
7 years ago
Kexin Zhao 4eaa789730 resolve conflict
7 years ago
Jacek Czaja 3b95b55f07 - Softmax MKLDNN primitive integration
7 years ago
Kexin Zhao b7801b9fcb small fix
7 years ago
Kexin Zhao 70e7122785 initial commit
7 years ago
dzhwinter 128adf53cb
[Speed]implement cudnn sequence softmax cudnn (#8978)
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