Commit Graph

4414 Commits (157211c4e14f2e2917f1098c6ad8dd0761050d14)

Author SHA1 Message Date
Yi Liu 157211c4e1
supports distributed classification (#18690)
6 years ago
qingqing01 3429e65aa8
Fix CPU implementation of roi_align_op backward (#18728)
6 years ago
Tao Luo bd22453f20
Revert "Add LeakyRelu MKLDNN support (#18656)" (#18723)
6 years ago
whs 189b08dc0d
Make infer shape of pad2d support for input with negative dims in compile time. (#18695)
6 years ago
Bai Yifan 7e3963f295 add license, test=develop (#18709)
6 years ago
cjt222 ccf06a48b0
test=develop (#18701)
6 years ago
wangguanzhong 185b3acea1
fix clip_by_norm doc (#18688)
6 years ago
Huihuang Zheng 89bc3fd841
Support memory eager deletion on recurrent OP (#17710)
6 years ago
Adam d6b6a337a9 Add LeakyRelu MKLDNN support (#18656)
6 years ago
hutuxian bb2f5d24a2
hash_op support int64 hash_size (#18674)
6 years ago
guru4elephant 5ed713d519
remove ctr reader, all functions are satisfied in dataset (#18672)
6 years ago
Yang Zhang ce1ec33299
Add cuda implementation for `prelu` backward pass (#18633)
6 years ago
Yihua Xu 97549a4f13 [CPU] Fix the compiling issue with AVX512F macro. (#18634)
6 years ago
baojun 256ba7cbb8 [NGraph] handle dim element 0 of ngraph op (#18568)
6 years ago
Jacek Czaja 71d883b8ef [MKL-DNN] Reimplemented pool2d mkl-dnn to use Acquire API (#18585)
6 years ago
chengduo f4ec7d54c8
fix bug of scatter op (#18640)
6 years ago
guru4elephant ab57d3893e
make auc op compatible with 1 dim (#18551)
7 years ago
Hongyu Liu a20b2b43fc
fix cudnn lstm shape bug; test=develop (#18492)
7 years ago
Zeng Jinle d3003a1620
Feature/buffer_shared_inplace (#17911)
7 years ago
Zeng Jinle be24e5b391
Clean unused code of dim and place (#18565)
7 years ago
Jacek Czaja 8869d7f735 Activations MKLDNN ops refactoring (#18191)
7 years ago
Yibing Liu b86234fc0b
Register fp16 for concat_op (#18563)
7 years ago
Physher 5e1220ef37 fix compile error which caused by gcc4.8 related commit;test=develop (#18567)
7 years ago
Jiabin Yang 667f88f9a6
Fix/gcc 4.8 ubt link error (#18558)
7 years ago
Physher 0caa08ea40 Add mkldnn int8 mul-op kernel (#17834)
7 years ago
LielinJiang 24d1c44a0c Fix roi_perspective_transform_op bug (#18522)
7 years ago
Zhaolong Xing 88b52a27fe
Inference: fix mask rcnn model diff, optim memory usage, memory leak. (#18532)
7 years ago
zhaoyuchen2018 832d8191ff
Fix topk cannot handle 1D vector bug (#18466)
7 years ago
qingqing01 7ac4818a98
Refine Infershape in activation_op for double_grad. (#18485)
7 years ago
chengduo 7453857324 Make fuse_all_reduce_op_pass support mix_precision (#17652)
7 years ago
zhoukunsheng 7c6f2350b9 support Tensor input for edit_distance op (#18162)
7 years ago
zhoukunsheng 26318544d2 support Tensor input for chunk_eval op (#18226)
7 years ago
zhoukunsheng 206c44e2a8 add unique kernel and op (#17557)
7 years ago
zhoukunsheng 71af72b1c2 upgrade hash op to support Tensor and LoDTensor input (#17998)
7 years ago
zhoukunsheng d3b3443d10 add ones_like op (#17388)
7 years ago
zhoukunsheng 67b48d7fe7 add size op (#17412)
7 years ago
Leo Zhao 8f5fffca0a rename mkldnn set/get_cur_thread_id() to set/get_cur_mkldnn_session_id() (#18453)
7 years ago
Yi Liu a873fa84ce
supports collective training with programs (#18392)
7 years ago
chengduo e0d8c6ac68
Add find_no_grad_vars in backward.py (#17942)
7 years ago
LielinJiang 449c7a9f98 Make roi_perspective_transform op return mask and transform matrix (#18371)
7 years ago
Brian Liu 4bc2987d2f Fix bug in quantize kernel which cause crash in vgg16/19 model (#17964)
7 years ago
Leo Zhao 681d3553f1 Fix potential mkldnn concat/pool/conv kernel issues (#18393)
7 years ago
Zeng Jinle f5641000bb
Add a unittest to inplace elementwise_add (#18385)
7 years ago
tangwei12 999d9a59a5
fix communicator with pyreader (#18350)
7 years ago
HaoRen b7128bac5f supports collective communicated training (#18175)
7 years ago
Sylwester Fraczek 9252e8fa08 add int8 mkldnn prior_box (#17242)
7 years ago
Jacek Czaja c2efdfd5bc [MKL-DNN] Extending reusing to Elementwise_add_mkldnn op (#18146)
7 years ago
qingqing01 9047ac687e
Simplify multi_box_head API in detection.py and remove assign op. (#18310)
7 years ago
Yibing Liu 23941e43ec
Update lamb optimizer (#18333)
7 years ago
tensor-tang 81ec538279
fix softrelu doc (#18324)
7 years ago