Commit Graph

4604 Commits (7836f477c6a5c7aac386f512f27399148aa9921d)

Author SHA1 Message Date
wangchaochaohu 3409db950c
fix reduce bug test=develop (#19971)
6 years ago
Adam 4b65af7719 MKLDNN BatchNorm operator refactor (#20012)
6 years ago
joanna.wozna.intel 1d32897c5c Fix test pool2d int8 mkldnn (#19976)
6 years ago
Aurelius84 f58c8db668
Require x.dims=label.dims in huber_loss (#20017)
6 years ago
Aurelius84 137e6336ef
Remove constraint that last dimension is forced to be 1 in rank_loss (#19997)
6 years ago
chengduo 101a2b610a Add dtype for coalesce_tensor_op (#20016)
6 years ago
Zhaolong Xing f04f2b232a
fix if else error info (#19974)
6 years ago
gongweibao a7512db2bc
Polish elementwise max min pow document to add more examples. (#19946)
6 years ago
Aurelius84 2b5b4b3c5e
fix dataType in C++ comment in embedding op (#20004)
6 years ago
Tao Luo bcb2903e60
enhance shape error message of mul_op (#19998)
6 years ago
Chen Weihang 1409586eaa Add LoD empty check for all related sequence ops (#19980)
6 years ago
zhongpu b1bb23841e add kernel for fill_op, test=develop (#19719)
6 years ago
wangchaochaohu 382d099dcb
add support tensor and tensorlist for strided_slice OP (#19929)
6 years ago
lvmengsi 619a241bd0
Fix OpTest of bn (#19062)
6 years ago
Bob Zhu c670058a8d add support of matmul with multiple head even different width and height (#19708)
6 years ago
Liufang Sang 6884dc800a refine ctc align op with padding (#19926)
6 years ago
Aurelius84 99a9615a4b
Removing length dims constraints of seq_pad and seq_unpad (#19497)
6 years ago
jhjiangcs 766bd529d1 add optimizer:dpsgd,test=develop (#19915)
6 years ago
Yang Zhang ebff68fa74
Add float16 support to `sync_batch_norm_op` (#19681)
6 years ago
Aurelius84 039b9710d5
Remove constraint that last dimension is forced to be 1 by adding lookup_table_v2 (#19735)
6 years ago
xujiaqi01 cedc04775c
support change shuffle and train thread num (#19841)
6 years ago
Kaipeng Deng 14625ffe9e
add elementwise mod support float/double. test=develop (#19570)
6 years ago
Jacek Czaja 5b07ca9cdd - ReImplemented pooling fwd mkldnn (#19911)
6 years ago
Zeng Jinle b1e83b33b0
fix huber loss op attr type, test=develop (#19937)
6 years ago
Zeng Jinle cc157d5990
add inplace to assign op, test=develop (#19927)
6 years ago
Leo Chen 57606205f5 Make OpTest check grad inplace even if forward has no inplace (#19847)
6 years ago
Zhang Ting cb8f3c03a7 resize Ops support data_layout:channel_last, test=develop, test=document_preview (#19914)
6 years ago
Kaipeng Deng 3f021781a1
fix softmax CE time limit check failed (#19846)
6 years ago
石晓伟 30adea0a23
tensor_array_to_tensor_op.cc, test=develop (#19289)
6 years ago
lvmengsi 4155e62559
add instance norm (#19500)
6 years ago
Adam cb65439da8 Add support for other axes in MKLDNN softmax op (#19907)
6 years ago
Pei Yang baccd7e2ca
Add TRT input shape check between model and runtime (#19864)
6 years ago
Aurelius84 fcf53e55ff
support 2-level lod of input in sequence_pool (#19839)
6 years ago
Zhang Ting 93364b45c1 group_norm support data_layout:NHWC, test=develop, test=document_preview (#19614)
6 years ago
Jacek Czaja 619c797a7f [MKL-DNN] LRN refactoring (#19798)
6 years ago
Zhang Ting 439d95e157 modified interpolate op to support tensor attribute, test=develop, test=document_preview (#19287)
6 years ago
Zhang Ting b38889413d add crop_tensor_op, test=develop, test=document_preview (#19314)
6 years ago
lidanqing 2c32c2d649 Refactor conv computeINT8 (#19574)
6 years ago
Adam c7e688921b Add template functions for Acquire primitive/primitive_desc (#19867)
6 years ago
Aurelius84 b125e327aa
Remove constraint that last dimension is forced to be 1 in cross_entropy (#19606)
6 years ago
wopeizl a7c440d303
add precise roi pooling op test=develop (#18960)
6 years ago
Yiqun Liu 3cd985a669
Add a pass to fuse fc+elementwise_add+layernorm (#19776)
6 years ago
wangchaochaohu 47af618f70
Strided slice (#19642)
6 years ago
123malin 1bc285a53a
add retry function to try to solve grpc error code 14 (#19661)
6 years ago
LielinJiang 6d72a86b14 fix_roi_transform_bug (#19785)
6 years ago
Zeng Jinle 3fd3b663a8
fix gc bug in controlflow ops, test=develop (#19827)
6 years ago
Leo Chen 982e61f5ff Update elementwise double grad to save gpu memory (#19509)
6 years ago
Adam dfdd73cbc0 Add MKLDNNhandlerT templatized class (#19801)
6 years ago
Zeng Jinle cabb9501bd
fix leaky_relu op when alpha is zero, test=develop (#19833)
6 years ago
chengjuntao 00efd1d8a9
add deformable conv v1 op and cpu version of deformable conv v2 (#18500)
6 years ago