Commit Graph

25 Commits (7f73c16e421fb2d181705838832153c3e8f92ec9)

Author SHA1 Message Date
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