ops.md 629 Bytes
Newer Older
Zaida Zhou's avatar
Zaida Zhou committed
1
2
## CUDA 算子

3
4
MMCV 提供了检测、分割等任务中常用的 CUDA 算子

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
- KNN
20
21
22
- MaskedConv
- NMS
- PSAMask
23
- RotatedFeatureAlign
24
- RoIPointPool3d
25
26
- RoIPool
- RoIAlign
27
- RoIAwarePool3d
28
29
30
31
32
- SimpleRoIAlign
- SigmoidFocalLoss
- SoftmaxFocalLoss
- SoftNMS
- Synchronized BatchNorm
33
- Voxelization
34
35
- ThreeInterpolate
- ThreeNN
36
- Weight standardization
37
- Correlation