ops.md 755 Bytes
Newer Older
1
2
3
4
## CUDA ops

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

5
- ActiveRotatedFilter
6
- AssignScoreWithK
7
- BallQuery
8
9
10
11
- BBoxOverlaps
- CARAFE
- CrissCrossAttention
- ContextBlock
12
- ConvexIoU
13
14
15
- CornerPool
- Deformable Convolution v1/v2
- Deformable RoIPool
16
- DynamicScatter
17
- GatherPoints
18
19
- FurthestPointSample
- FurthestPointSampleWithDist
20
- GeneralizedAttention
21
- GroupPoints
22
- KNN
23
- MaskedConv
24
- MinAreaPolygon
25
- NMS
26
- PointsInPolygons
27
- PSAMask
28
- RiRoIAlignRotated
29
- RotatedFeatureAlign
30
- RoIPointPool3d
31
32
- RoIPool
- RoIAlign
33
- RoIAwarePool3d
34
35
36
37
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
38
- Sparse Convolution
39
- Synchronized BatchNorm
40
- Voxelization
41
42
- ThreeInterpolate
- ThreeNN
43
- Weight standardization
44
- Correlation