Shortcuts

ops

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

Device CPU CUDA MLU MPS Ascend
ActiveRotatedFilter
AssignScoreWithK
BallQuery
BBoxOverlaps
BorderAlign
BoxIouRotated
BoxIouQuadri
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
NMSQuadri
PixelGroup
PointsInBoxes
PointsInPolygons
PSAMask
RotatedFeatureAlign
RoIPointPool3d
RoIPool
RoIAlignRotated
RiRoIAlignRotated
RoIAlign
RoIAwarePool3d
SAConv2d
SigmoidFocalLoss
SoftmaxFocalLoss
SoftNMS
Sparse Convolution
Synchronized BatchNorm
ThreeInterpolate
ThreeNN
TINShift
UpFirDn2d
Voxelization
PrRoIPool
BezierAlign
BiasAct
FilteredLrelu
Conv2dGradfix