mmcv/docs/en/understand_mmcv/ops.md

ops

We implement common ops used in detection, segmentation, etc.

Device	CPU	CUDA	MLU	MPS
ActiveRotatedFilter	√	√
AssignScoreWithK		√
BallQuery		√
BBoxOverlaps		√	√	√
BorderAlign		√
BoxIouRotated	√	√
CARAFE		√
ChamferDistance		√
CrissCrossAttention		√
ContourExpand	√
ConvexIoU		√
CornerPool		√
Correlation		√
Deformable Convolution v1/v2	√	√
Deformable RoIPool		√
DiffIoURotated		√
DynamicScatter		√
FurthestPointSample		√
FurthestPointSampleWithDist		√
FusedBiasLeakyrelu		√
GatherPoints		√
GroupPoints		√
Iou3d		√
KNN		√
MaskedConv		√
MergeCells		√
MinAreaPolygon		√
ModulatedDeformConv2d	√	√
MultiScaleDeformableAttn		√
NMS	√	√	√
NMSRotated	√	√
PixelGroup	√
PointsInBoxes	√	√
PointsInPolygons		√
PSAMask	√	√	√
RotatedFeatureAlign	√	√
RoIPointPool3d		√
RoIPool		√	√
RoIAlignRotated	√	√	√
RiRoIAlignRotated		√
RoIAlign	√	√	√
RoIAwarePool3d		√
SAConv2d		√
SigmoidFocalLoss		√	√
SoftmaxFocalLoss		√
SoftNMS		√
Sparse Convolution		√
Synchronized BatchNorm		√
ThreeInterpolate		√
ThreeNN		√
TINShift		√	√
UpFirDn2d		√
Voxelization	√	√