Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.md 3.39 KB
Newer Older
Alex Yang's avatar
[Docs] Reformat op docs (#2052)  
Alex Yang committed
1 
## 算子
Zaida Zhou's avatar
[Docs] Build Chinese docs (#1073)  
Zaida Zhou committed
2
Alex Yang's avatar
[Docs] Reformat op docs (#2052)  
Alex Yang committed
3 
MMCV 提供了检测、分割等任务中常用的算子
Zaida Zhou's avatar
[Docs] Translate ops documentation (#1349)  
Zaida Zhou committed
4
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30 
| 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                          |     | √    |     |     |
bdf's avatar
[Feature] Support MaskedConv2d with cambricon MLU backend (#2202)  
bdf committed
31 
| MaskedConv                   |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60 
| 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                 | √   | √    |     |     |
| PrRoIPool                    |     | √    |     |     |