Rui Xu 933b052d95

[Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu (#900) ...

* add upfirdn2d op

* fix bug in pybind

* add fused bias leakyrelu

* fix bug in fused-bias-leakyrelu

* fix lint error

* fix bug in build cpu version

* fix bug in build cpu version

* fix lint

* fix comment from zww

Co-authored-by: zhangshilong <zhangshilong@sensetime.com>

2021-03-21 21:23:09 +08:00

..

test_bbox.py

Misc updates regarding Windows and unittests (#586)

2020-10-28 16:23:19 +08:00

test_box_iou_rotated.py

add 'iof' mode for box_iou_rotated (#753)

2020-12-25 17:06:45 +08:00

test_carafe.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_cc_attention.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_corner_pool.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_deform_conv.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_deform_roi_pool.py

Add new parrots extension implementation for all ops (#794)

2021-02-26 19:05:25 +08:00

test_focal_loss.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_fused_bias_leakyrelu.py

[Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu (#900)

2021-03-21 21:23:09 +08:00

test_info.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_masked_conv2d.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_merge_cells.py

Migrate op (#392)

2020-07-08 17:29:15 +08:00

test_modulated_deform_conv.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_nms_rotated.py

Fix some bug as well as unit test for nms_rotate (#728)

2020-12-18 15:17:11 +08:00

test_nms.py

Add new parrots extension implementation for all ops (#794)

2021-02-26 19:05:25 +08:00

test_onnx.py

Add new parrots extension implementation for all ops (#794)

2021-02-26 19:05:25 +08:00

test_psa_mask.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_roi_align.py

fix roi_align ci for parrots (#708)

2020-12-13 20:01:25 +08:00

test_roi_pool.py

add ext ops, support parrots (#310)

2020-06-28 23:15:47 +08:00

test_saconv.py

fix saconv (#489)

2020-08-16 01:45:01 +08:00

test_syncbn.py

fix syncbn parameter order mismatch and parrots bug (#488)

2020-08-13 19:27:05 +08:00

test_tensorrt.py

[Feature] : Add Deformable Conv2d TensorRT Plugin (#858)

2021-03-11 15:41:47 +08:00

test_tin_shift.py

Add tin_shift function (#492)

2020-08-15 22:29:07 +08:00

test_upfirdn2d.py

[Feature] Add cuda ops: UpFirDn2d and fused_bias_leakyrelu (#900)

2021-03-21 21:23:09 +08:00