Commit Graph

4598 Commits (f04f2b232a22c9aba3ee4538ab708acf9f77c813)

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