Commit Graph

42 Commits (b4b6763ab20805c7447bf6485164d5d595bbb3ed)

Author SHA1 Message Date
Zeng Jinle d33c4343e1
Imperative tracer refactoring (#22457)
5 years ago
Zeng Jinle 878a40f57d
Support NoNeedBufferVarsInference in dygraph backward (#20868)
5 years ago
hong 8c4573a3cb
GradMaker for dygraph (#19706)
5 years ago
Chen Weihang 26cc1fe508
Replace risky GetInputType method with secure IndicateVarDataType interface (#20668)
5 years ago
Adam cb65439da8 Add support for other axes in MKLDNN softmax op (#19907)
5 years ago
Zeng Jinle 710767d894
Enable inplace support for some ops (#19612)
6 years ago
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