mmcv/tests/test_ops
v-qjqs 934b549e23
Support custom operators cummax and cummin for onnxruntime (#1010)
* support custom op `mmcv::cummax` for onnxruntime in mmcv

* fix clang-format lint error

* support mmcv::cummin, reformat codes

* fix merge from master

* add docs for mmcv::cummax and mmcv::cummin

* format doc

* add assertion for torch version, when exporting `cummax` to onnx

* add more comments for torch version

* handle exporting to onnx in `soft_nms`

* commit for test_onnx

* remove `is_in_onnx_export` in softnms

* add more comments

* fix c++ lint error

* add known issues doc for `cummax`

* fix known issues doc
2021-05-10 21:33:27 +08:00
..
test_bbox.py
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
test_deform_conv.py [Bug] Fix DeformConv2d bias error and add tests (#940) 2021-04-12 20:31:34 +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
test_fused_bias_leakyrelu.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00
test_info.py
test_masked_conv2d.py
test_merge_cells.py
test_modulated_deform_conv.py
test_ms_deformable_attn.py [Feature]: support Multi-Scale-DeformAttention in deformable-detr (#878) 2021-04-23 16:35:15 +08:00
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 Support custom operators cummax and cummin for onnxruntime (#1010) 2021-05-10 21:33:27 +08:00
test_psa_mask.py
test_roi_align.py
test_roi_align_rotated.py [Feature]: Add Rotated ROI align op for pytorch (cpu&cuda), parrots (cpu&cuda) and onnxruntime (cpu) (#933) 2021-04-19 15:04:01 +08:00
test_roi_pool.py
test_saconv.py
test_syncbn.py
test_tensorrt.py [Feature] : Add grid_sampler TensorRT Plugin (#865) 2021-03-31 00:28:08 +08:00
test_tin_shift.py
test_upfirdn2d.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00