Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ops.md 4.33 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
ckirchhoff's avatar
[Feature] Pick npu ops from master to 2.x (#2501)  
ckirchhoff committed
5
6
7
8
9 
| Device                       | CPU | CUDA | MLU | MPS | Ascend |
| ---------------------------- | --- | ---- | --- | --- | ------ |
| ActiveRotatedFilter          | √   | √    |     |     |        |
| AssignScoreWithK             |     | √    |     |     |        |
| BallQuery                    |     | √    |     |     |        |
ckirchhoff's avatar
[Feature] Add support of some ops for Ascend device (#2594)  
ckirchhoff committed
10 
| BBoxOverlaps                 |     | √    | √   | √   | √      |
ckirchhoff's avatar
[Feature] Pick npu ops from master to 2.x (#2501)  
ckirchhoff committed
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27 
| BorderAlign                  |     | √    |     |     |        |
| BoxIouRotated                | √   | √    |     |     |        |
| BoxIouQuadri                 | √   | √    |     |     |        |
| CARAFE                       |     | √    | √   |     |        |
| ChamferDistance              |     | √    |     |     |        |
| CrissCrossAttention          |     | √    |     |     |        |
| ContourExpand                | √   |      |     |     |        |
| ConvexIoU                    |     | √    |     |     |        |
| CornerPool                   |     | √    |     |     |        |
| Correlation                  |     | √    |     |     |        |
| Deformable Convolution v1/v2 | √   | √    |     |     | √      |
| Deformable RoIPool           |     | √    | √   |     | √      |
| DiffIoURotated               |     | √    |     |     |        |
| DynamicScatter               |     | √    |     |     |        |
| FurthestPointSample          |     | √    |     |     |        |
| FurthestPointSampleWithDist  |     | √    |     |     |        |
| FusedBiasLeakyrelu           |     | √    |     |     | √      |
ckirchhoff's avatar
[Feature] Add support of some ops for Ascend device (#2594)  
ckirchhoff committed
28 
| GatherPoints                 |     | √    |     |     | √      |
ckirchhoff's avatar
[Feature] Pick npu ops from master to 2.x (#2501)  
ckirchhoff committed
29
30
31
32
33
34
35
36
37 
| GroupPoints                  |     | √    |     |     |        |
| Iou3d                        |     | √    | √   |     |        |
| KNN                          |     | √    |     |     |        |
| MaskedConv                   |     | √    | √   |     | √      |
| MergeCells                   |     | √    |     |     |        |
| MinAreaPolygon               |     | √    |     |     |        |
| ModulatedDeformConv2d        | √   | √    |     |     | √      |
| MultiScaleDeformableAttn     |     | √    | √   |     |        |
| NMS                          | √   | √    | √   |     | √      |
ckirchhoff's avatar
[Feature] Add support of some ops for Ascend device (#2594)  
ckirchhoff committed
38 
| NMSRotated                   | √   | √    |     |     | √      |
ckirchhoff's avatar
[Feature] Pick npu ops from master to 2.x (#2501)  
ckirchhoff committed
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60 
| NMSQuadri                    | √   | √    |     |     |        |
| PixelGroup                   | √   |      |     |     |        |
| PointsInBoxes                | √   | √    |     |     |        |
| PointsInPolygons             |     | √    |     |     |        |
| PSAMask                      | √   | √    | √   |     | √      |
| RotatedFeatureAlign          | √   | √    |     |     |        |
| RoIPointPool3d               |     | √    | √   |     |        |
| RoIPool                      |     | √    | √   |     | √      |
| RoIAlignRotated              | √   | √    | √   |     |        |
| RiRoIAlignRotated            |     | √    |     |     |        |
| RoIAlign                     | √   | √    | √   |     |        |
| RoIAwarePool3d               |     | √    | √   |     |        |
| SAConv2d                     |     | √    |     |     |        |
| SigmoidFocalLoss             |     | √    | √   |     | √      |
| SoftmaxFocalLoss             |     | √    |     |     | √      |
| SoftNMS                      |     | √    |     |     |        |
| Sparse Convolution           |     | √    |     |     |        |
| Synchronized BatchNorm       |     | √    |     |     |        |
| ThreeInterpolate             |     | √    |     |     |        |
| ThreeNN                      |     | √    | √   |     |        |
| TINShift                     |     | √    | √   |     |        |
| UpFirDn2d                    |     | √    |     |     |        |
liuhw's avatar
[Feature] Add the support of voxelization op for ascend device (#2614)  
liuhw committed
61 
| Voxelization                 | √   | √    |     |     | √      |
ckirchhoff's avatar
[Feature] Pick npu ops from master to 2.x (#2501)  
ckirchhoff committed
62
63 
| PrRoIPool                    |     | √    |     |     |        |
| BezierAlign                  | √   | √    |     |     |        |
Yifei Yang's avatar
[Feature] Add Ops of StyleGAN3 (#2290)  
Yifei Yang committed
64
65
66 
| BiasAct                      |     | √    |     |     |        |
| FilteredLrelu                |     | √    |     |     |        |
| Conv2dGradfix                |     | √    |     |     |        |