mmcv/tests/test_ops
q.yao 9d80f56ae8
[Feature] : Add grid_sampler TensorRT Plugin (#865)
* add grid sample trt support

* fix align_corners=True

* change 'intep' to 'interp', fix bugs of grid sampler, better test script

* remove unused import

* move source comment to the top of trt_grid_sampler_kernel.cu
2021-03-31 00:28:08 +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
test_cc_attention.py
test_corner_pool.py Migrate op (#392) 2020-07-08 17:29:15 +08:00
test_deform_conv.py
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
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
test_masked_conv2d.py
test_merge_cells.py Migrate op (#392) 2020-07-08 17:29:15 +08:00
test_modulated_deform_conv.py
test_nms.py Add new parrots extension implementation for all ops (#794) 2021-02-26 19:05:25 +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_onnx.py Add new parrots extension implementation for all ops (#794) 2021-02-26 19:05:25 +08:00
test_psa_mask.py
test_roi_align.py fix roi_align ci for parrots (#708) 2020-12-13 20:01:25 +08:00
test_roi_pool.py
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 grid_sampler TensorRT Plugin (#865) 2021-03-31 00:28:08 +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