Commit Graph

79 Commits (b34933d9ee3b61dbbd642fd02f244c36d0d14550)

Author SHA1 Message Date
Adam d4413a54bc Add common CreateKey for mkldnn handlers (#19767)
6 years ago
Adam 428b2b9e17 MKLDNN handler cleanup (#19713)
6 years ago
Tao Luo f05d2c519d paddle::framework::vectorize() templatization [PART3] (#19643)
6 years ago
danleifeng 8672e15363 elementwise broadcast function enhancement (#19536)
6 years ago
Adam e94b26daf5 using MKLDNNMemoryFormat = mkldnn::memory::format changes (#19568)
6 years ago
Zeng Jinle 5c8f210ce3
refine inplace inference registry, test=develop (#19032)
6 years ago
zhaoyuchen2018 5296294dae
Fix elementwise performance poor issue (#19278)
6 years ago
danleifeng e0a2d4dfec Add elementwise_pow_op backward implementation and the unit test codes of it. (#18848)
6 years ago
Zeng Jinle f5641000bb
Add a unittest to inplace elementwise_add (#18385)
6 years ago
Jacek Czaja c2efdfd5bc [MKL-DNN] Extending reusing to Elementwise_add_mkldnn op (#18146)
6 years ago
Yiqun Liu 660c1a65f3
Optimize fused_elewise_activation_grad op. (#18041)
6 years ago
Tao Luo 53fd507bae
fix merge conflict of 'Remove attribute in Allocator::Allocate' and elementwise_add_mkldnn_op (#17949)
6 years ago
Yihua Xu 9b5017366a Fix the format issue when 'X' is not nchw. (#17833)
6 years ago
lvmengsi 10b23a72c1 Double backward elementwise div (#17416)
6 years ago
lvmengsi 977e9fcb27
support elementwise_sub double backward (#17476)
6 years ago
Kaipeng Deng bd9bef5a4e
add elementwise_add_grad_grad op (#17366)
6 years ago
Yiqun Liu dcda20233c
Optimize the elementwise op using eigen (#15494)
6 years ago
Kaipeng Deng 8bae8590ac
add double grad for elementwise_mul op (#17255)
6 years ago
zhoukunsheng 4292bd8687 Mod floordiv (#17251)
6 years ago
zhaoyuchen2018 792443ef23
Refine elementwise kernel. (#16952)
6 years ago
gongweibao 91784f8ec3
Fix code in document. (#17237)
6 years ago
Zeng Jinle ee2028a110
Add use_cuda to inplace pass (#17205)
6 years ago
Zeng Jinle 1c526e1d1a
Fix some grad op desc makers (#16633)
6 years ago
Jacek Czaja 2632327429 [MKL-DNN] Tensor modifications revert (#16462)
6 years ago
Zeng Jinle c7c6eeb44e
Merge pull request #16409 from sneaxiy/feature/advance_gc
6 years ago
liuwei1031 8d22bc17a4
Memory optimize (#16410)
6 years ago
sneaxiy 072d95d8f6 Merge develop
6 years ago
sneaxiy a93a9eef8f add op registry type
6 years ago
phlrain 5dc9b51994 fix time; test=develop
6 years ago
phlrain 18d107c27a add floordiv and mod op; test=develop
6 years ago
phlrain 56c2d384c7 add elementwise floordiv, mod; test=develop
6 years ago
tensor-tang 14a764c930 simplify the jitkernel templates and tests
6 years ago
tensor-tang 802f362ac4 unify the kernelfuncs cache and add unit test
6 years ago
Jacek Czaja c63f6b2039 - MKL-DNN pooling updated to set_prim_desc
6 years ago
dzhwinter 104d3b4e68 add details. test=develop
6 years ago
dzhwinter 94dd50c33f add details. test=develop
6 years ago
dzhwinter 9c9ad7d40b Merge remote-tracking branch 'origin/develop' into feature/ir_inplace_pass
6 years ago
Krzysztof Binias 69b7c595d6 Small fix
6 years ago
Krzysztof Binias b1bdcd4de8 Make separate folders for mkldnn codes
6 years ago
dzhwinter ee3aae56cd merge develop branch. test=develop
6 years ago
chengduo bf91d11ed5
Clean elementwise_op_function (#15502)
6 years ago
dzhwinter 8f3b252392 squash commits. test=develop
6 years ago
Xin Pan 3ecf6bb338
Merge pull request #15028 from yihuaxu/develop_641313ea7_elementwise_mul_mkldnn_bug_fix
6 years ago
Wu Yi fd85418329
[Feature] support mix precision training for resnet (#14899)
6 years ago
sneaxiy 73896eeb94 merge develop
6 years ago
sneaxiy b56aca82e9 merge develop
6 years ago
sneaxiy dde3afe7b7 Merge develop
6 years ago
Wu Yi 856f0da0fe
Fp16 training (#14992)
6 years ago
Yihua Xu d4606bcb22 Fix the exception when tensor format is x
6 years ago
peizhilin 2e35290ffa fix build issue
6 years ago