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

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

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