mmcv/tests/test_ops
v-qjqs db6b0542c7
[Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997)
* supports for onnxruntime custom op `mmcv::MMCVTopPool`

* supports for onnxruntime custom op `mmcv::MMCVCornerPool`, involving TopPool, BottomPool, LeftPool and RightPool

* add unittest for corner_pool

* supports mmcv::CornerPool without memcpy

* add docs for mmcv::CornerPool

* re-add docs for mmcv::CornerPool

* fix output dtype doc

* reformat

* format with pre-commit

* format

* fix lint error, by using google clang-format style for c/c++
2021-05-01 16:28:25 +08:00
..
test_bbox.py
test_box_iou_rotated.py
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
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
test_nms_rotated.py
test_onnx.py [Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997) 2021-05-01 16:28:25 +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
test_tin_shift.py
test_upfirdn2d.py [Feature]: parrots add parrots/fused_bias & upfirdn2d (#989) 2021-04-27 13:49:50 +08:00