Skip to content

GitLab

"src/targets/gpu/vscode:/vscode.git/clone" did not exist on "e4f5508c134e83225cb43a8518f41af49384d9a6"
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
q.yao's avatar
[Feature] Add MPS bbox overlap (#2123)  
q.yao committed
5
6
7
8
9
10
11
12 
| Device                       | CPU | CUDA | MLU | MPS |
| ---------------------------- | --- | ---- | --- | --- |
| ActiveRotatedFilter          | √   | √    |     |     |
| AssignScoreWithK             |     | √    |     |     |
| BallQuery                    |     | √    |     |     |
| BBoxOverlaps                 |     | √    | √   | √   |
| BorderAlign                  |     | √    |     |     |
| BoxIouRotated                | √   | √    |     |     |
bdf's avatar
[Feature] Add carafe op for MLU (#2212)  
bdf committed
13 
| CARAFE                       |     | √    | √   |     |
q.yao's avatar
[Feature] Add MPS bbox overlap (#2123)  
q.yao committed
14
15
16
17
18
19
20 
| ChamferDistance              |     | √    |     |     |
| CrissCrossAttention          |     | √    |     |     |
| ContourExpand                | √   |      |     |     |
| ConvexIoU                    |     | √    |     |     |
| CornerPool                   |     | √    |     |     |
| Correlation                  |     | √    |     |     |
| Deformable Convolution v1/v2 | √   | √    |     |     |
bdf's avatar
[Feature] Support DeformRoiPool with cambricon MLU backend (#2137)  
bdf committed
21 
| Deformable RoIPool           |     | √    | √   |     |
q.yao's avatar
[Feature] Add MPS bbox overlap (#2123)  
q.yao committed
22
23
24
25
26
27
28
29
30 
| DiffIoURotated               |     | √    |     |     |
| DynamicScatter               |     | √    |     |     |
| FurthestPointSample          |     | √    |     |     |
| FurthestPointSampleWithDist  |     | √    |     |     |
| FusedBiasLeakyrelu           |     | √    |     |     |
| GatherPoints                 |     | √    |     |     |
| GroupPoints                  |     | √    |     |     |
| Iou3d                        |     | √    |     |     |
| KNN                          |     | √    |     |     |
bdf's avatar
[Feature] Support MaskedConv2d with cambricon MLU backend (#2202)  
bdf committed
31 
| MaskedConv                   |     | √    | √   |     |
q.yao's avatar
[Feature] Add MPS bbox overlap (#2123)  
q.yao 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 
| 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             |     | √    |     |     |
liuduanhui's avatar
[Feature] Support ThreeNN with cambricon MLU backend (#2215)  
liuduanhui committed
56 
| ThreeNN                      |     | √    | √   |     |
q.yao's avatar
[Feature] Add MPS bbox overlap (#2123)  
q.yao committed
57
58
59 
| TINShift                     |     | √    | √   |     |
| UpFirDn2d                    |     | √    |     |     |
| Voxelization                 | √   | √    |     |     |
Jingwei Zhang's avatar
[Feature] Support PrRoIPool operation (#2113)  
Jingwei Zhang committed
60 
| PrRoIPool                    |     | √    |     |     |