- 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 3 次提交
-
-
由 Chengmo 提交于
* fix zip py3 bug
-
由 Kaipeng Deng 提交于
-
由 Zhou Wei 提交于
-
- 24 3月, 2021 7 次提交
-
-
由 winter-wang 提交于
-
由 parap1uie-s 提交于
Fix default argument of nn.MaxPool3D()
-
由 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.
-
由 Aurelius84 提交于
-
由 cc 提交于
* Refine saving output scale to infer program
-
由 Chen Weihang 提交于
-
由 ronnywang 提交于
-
- 23 3月, 2021 4 次提交
-
-
由 Feiyu Chan 提交于
-
由 Wojciech Uss 提交于
-
由 tianshuo78520a 提交于
* Delete fast_check_nan_inf * Delete run_fast_nan_inf_debug
-
由 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 2 次提交
- 19 3月, 2021 7 次提交
-
-
由 Jacek Czaja 提交于
-
由 Chen Weihang 提交于
* support float16 in custom op * fix failed unittests
-
由 ronnywang 提交于
-
由 Chen Weihang 提交于
* support attribute in infershape * polish details
-
由 Adam Osewski 提交于
-
由 cc 提交于
* Refine calculating output scale of dygraph qat, test=develop
-
由 ronnywang 提交于
-
- 18 3月, 2021 5 次提交
-
-
由 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
-
由 Zhou Wei 提交于
-
由 Chengmo 提交于
* fix one ps gradient clip
-
由 Kaipeng Deng 提交于
* remove useless import. test=develop
-
- 17 3月, 2021 1 次提交
-
-
由 Zhang Ting 提交于
-