- 29 3月, 2021 5 次提交
-
-
由 liym27 提交于
-
由 ronnywang 提交于
-
由 Jiabin Yang 提交于
* give shape related contructor and reshape warning * change line num to fit ut * change ut to fit * remove useless code * call resize directly in constructor
-
由 zlsh80826 提交于
* add roi_align_plugin * add roi align unit_test * add roi align serialization * remove roi align static plugin because of batch dim issue * refine roi align unittest and add fp16/serialization * add trt roi align condition to op_teller * refine error message * remove unnecessary reshape layer
-
由 zlsh80826 提交于
* trt affine channel converter * add trt affine channel base test * add trt affine channel NHWC * remove asterisk for python2 compatibility * trt affine channel converter * add trt affine channel base test * add trt affine channel NHWC * remove asterisk for python2 compatibility * fix rebase * move LodTensor to Tensor * add dbg info * affine channel converter only support NCHW * scale,bias are parameters, use create_parameters api * reduce test input size to not exceed the timelimit of ci * refine affine channel unittest and add serialization/dynamic test * change super to InferencePassTest for python2 compatibility * change super to InferencePassTest for python2 compatibility * fix affine channel fp16 serialize setting
-
- 26 3月, 2021 3 次提交
-
-
由 cc 提交于
* Use layer to calculate output scale * add backward for moving_average_abs_max_scale and save output scales to op's attr
-
由 lilong12 提交于
* update, test=develop
-
由 zlsh80826 提交于
* add multiclass_nms * add multiclass_nms unittest * add default enable_tensorrt_oss option * refine multiclas nms unittest and add serialization/dynamic test * change super to InferencePassTest for python2 compatibility * refine multiclass nms unittest * move out dynamic shape test due to ci timelimit
-
- 25 3月, 2021 2 次提交
- 24 3月, 2021 2 次提交
-
-
由 Huihuang Zheng 提交于
Our old `loop_body` function may return single element when `loop_vars` just contains only 1 element, which can cause bug. The key point of this PR is forcing `loop_body` functions always return tuple.
-
由 ronnywang 提交于
-
- 23 3月, 2021 2 次提交
-
-
由 Feiyu Chan 提交于
-
由 gongweibao 提交于
fix launch ps ut test=develop
-
- 22 3月, 2021 3 次提交
-
-
由 zlsh80826 提交于
* nearest_interp op converter w/ dynamic/static * fix data_layout include * add trt nearest unit_test * add nearest_interp NHWC test * update trt nearest interp nhwc testcase * remove asterisk for python2 compatibility * add empty line to prevent conflict * nearest_interp op converter w/ dynamic/static * fix data_layout include * add trt nearest unit_test * add nearest_interp NHWC test * update trt nearest interp nhwc testcase * remove asterisk for python2 compatibility * add empty line to prevent conflict * change the priority of out_h, out_w
-
由 arlesniak 提交于
-
由 lilong12 提交于
* add 1f1b scheduler for pp, test=develop
-
- 21 3月, 2021 1 次提交
-
-
由 ronnywang 提交于
-
- 19 3月, 2021 6 次提交
-
-
由 Jacek Czaja 提交于
-
由 Chen Weihang 提交于
* support float16 in custom op * fix failed unittests
-
由 ronnywang 提交于
-
由 Chen Weihang 提交于
* support attribute in infershape * polish details
-
由 Adam Osewski 提交于
-
由 ronnywang 提交于
-
- 18 3月, 2021 3 次提交
-
-
由 Chen Weihang 提交于
* support custom complex op * fix detail error * add inference support * fix setup windows failed
-
由 zlsh80826 提交于
* trt gather converter * add trt gather unit_test
-
由 Chengmo 提交于
* fix one ps gradient clip
-
- 17 3月, 2021 1 次提交
-
-
由 Zhang Ting 提交于
-
- 16 3月, 2021 2 次提交
-
-
由 gongweibao 提交于
-
由 ronnywang 提交于
-
- 15 3月, 2021 2 次提交
-
-
由 Kaipeng Deng 提交于
* add dict/str/list supprot for DataLoader. test=develop
-
由 Chen Weihang 提交于
-
- 12 3月, 2021 4 次提交
-
-
由 Shang Zhizhou 提交于
* add serialize unittest * fix element_op trt plugin serialize bug
-
由 hong 提交于
-
由 Qi Li 提交于
-
由 Chen Weihang 提交于
* support duplicable op inout * add costom concat op test
-
- 11 3月, 2021 3 次提交
-
-
由 Aurelius84 提交于
* Fix bug with static_convert_var_shape * replace dot with dash
-
由 chajchaj 提交于
* add softmax_switch for softmax_with_cross_entropy_op, test=develop * delete using EigenMatrix in softmax_with_cross_entropy_op.h, test=develop * add REGISTER_OP_VERSION for softmax_switch attr of softmax_with_cross_entropy_op, test=develop
-
由 Aurelius84 提交于
* fix bug with jit.save * refine code
-
- 10 3月, 2021 1 次提交
-
-
由 WangXi 提交于
-