Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.md 3.52 KB
Newer Older
Alex Yang's avatar
[Docs] Reformat op docs (#2052)  
Alex Yang committed
1 
## ops
zhuyuanhao's avatar
add ext ops, support parrots (#310)  
zhuyuanhao committed
2
Alex Yang's avatar
[Docs] Reformat op docs (#2052)  
Alex Yang committed
3 
We implement common ops used in detection, segmentation, etc.
zhuyuanhao's avatar
add ext ops, support parrots (#310)  
zhuyuanhao committed
4
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
5
6
7
8
9
10
11
12 
| Device                       | CPU | CUDA | MLU | MPS |
| ---------------------------- | --- | ---- | --- | --- |
| ActiveRotatedFilter          | √   | √    |     |     |
| AssignScoreWithK             |     | √    |     |     |
| BallQuery                    |     | √    |     |     |
| BBoxOverlaps                 |     | √    | √   | √   |
| BorderAlign                  |     | √    |     |     |
| BoxIouRotated                | √   | √    |     |     |
Yue Zhou's avatar
[Feature] Add box_iou_quadri & nms_quadri (#2277)  
Yue Zhou committed
13 
| BoxIouQuadri                 | √   | √    |     |     |
bdf's avatar
[Feature] Add carafe op for MLU (#2212)  
bdf committed
14 
| CARAFE                       |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
15
16
17
18
19
20
21 
| ChamferDistance              |     | √    |     |     |
| CrissCrossAttention          |     | √    |     |     |
| ContourExpand                | √   |      |     |     |
| ConvexIoU                    |     | √    |     |     |
| CornerPool                   |     | √    |     |     |
| Correlation                  |     | √    |     |     |
| Deformable Convolution v1/v2 | √   | √    |     |     |
bdf's avatar
[Feature] Support DeformRoiPool with cambricon MLU backend (#2137)  
bdf committed
22 
| Deformable RoIPool           |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
23
24
25
26
27
28
29
30
31 
| DiffIoURotated               |     | √    |     |     |
| DynamicScatter               |     | √    |     |     |
| FurthestPointSample          |     | √    |     |     |
| FurthestPointSampleWithDist  |     | √    |     |     |
| FusedBiasLeakyrelu           |     | √    |     |     |
| GatherPoints                 |     | √    |     |     |
| GroupPoints                  |     | √    |     |     |
| Iou3d                        |     | √    |     |     |
| KNN                          |     | √    |     |     |
bdf's avatar
[Feature] Support MaskedConv2d with cambricon MLU backend (#2202)  
bdf committed
32 
| MaskedConv                   |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
33
34
35
36
37
38 
| MergeCells                   |     | √    |     |     |
| MinAreaPolygon               |     | √    |     |     |
| ModulatedDeformConv2d        | √   | √    |     |     |
| MultiScaleDeformableAttn     |     | √    |     |     |
| NMS                          | √   | √    | √   |     |
| NMSRotated                   | √   | √    |     |     |
Yue Zhou's avatar
[Feature] Add box_iou_quadri & nms_quadri (#2277)  
Yue Zhou committed
39 
| NMSQuadri                    | √   | √    |     |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
40
41
42
43
44 
| PixelGroup                   | √   |      |     |     |
| PointsInBoxes                | √   | √    |     |     |
| PointsInPolygons             |     | √    |     |     |
| PSAMask                      | √   | √    | √   |     |
| RotatedFeatureAlign          | √   | √    |     |     |
ZShaopeng's avatar
[Feature] Support RoipointPool3d with cambricon MLU backend (#2247)  
ZShaopeng committed
45 
| RoIPointPool3d               |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
46
47
48
49
50
51
52
53
54
55
56
57 
| 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
58 
| ThreeNN                      |     | √    | √   |     |
Zaida Zhou's avatar
Pick commits from master (#2164)  
Zaida Zhou committed
59
60
61
62 
| TINShift                     |     | √    | √   |     |
| UpFirDn2d                    |     | √    |     |     |
| Voxelization                 | √   | √    |     |     |
| PrRoIPool                    |     | √    |     |     |