mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Docs] Add currently npu supported ops in ops.md (#2420)

pull/2423/head
ckirchhoff 2022-11-17 14:16:36 +08:00 committed by GitHub
parent eda3c57ea1
commit 944a53f9d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 116 additions and 116 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

116
docs/en/understand_mmcv/ops.md
View File

@ -2,61 +2,61 @@
We implement common ops used in detection, segmentation, etc. We implement common ops used in detection, segmentation, etc.
| Device | CPU | CUDA | MLU | MPS | | Device | CPU | CUDA | MLU | MPS | NPU |
| ---------------------------- | --- | ---- | --- | --- | | ---------------------------- | --- | ---- | --- | --- | --- |
| ActiveRotatedFilter | √ | √ | | | | ActiveRotatedFilter | √ | √ | | | |
| AssignScoreWithK | | √ | | | | AssignScoreWithK | | √ | | | |
| BallQuery | | √ | | | | BallQuery | | √ | | | |
| BBoxOverlaps | | √ | √ | √ | | BBoxOverlaps | | √ | √ | √ | |
| BorderAlign | | √ | | | | BorderAlign | | √ | | | |
| BoxIouRotated | √ | √ | | | | BoxIouRotated | √ | √ | | | |
| BoxIouQuadri | √ | √ | | | | BoxIouQuadri | √ | √ | | | |
| CARAFE | | √ | √ | | | CARAFE | | √ | √ | | |
| ChamferDistance | | √ | | | | ChamferDistance | | √ | | | |
| CrissCrossAttention | | √ | | | | CrissCrossAttention | | √ | | | |
| ContourExpand | √ | | | | | ContourExpand | √ | | | | |
| ConvexIoU | | √ | | | | ConvexIoU | | √ | | | |
| CornerPool | | √ | | | | CornerPool | | √ | | | |
| Correlation | | √ | | | | Correlation | | √ | | | |
| Deformable Convolution v1/v2 | √ | √ | | | | Deformable Convolution v1/v2 | √ | √ | | | |
| Deformable RoIPool | | √ | √ | | | Deformable RoIPool | | √ | √ | | |
| DiffIoURotated | | √ | | | | DiffIoURotated | | √ | | | |
| DynamicScatter | | √ | | | | DynamicScatter | | √ | | | |
| FurthestPointSample | | √ | | | | FurthestPointSample | | √ | | | |
| FurthestPointSampleWithDist | | √ | | | | FurthestPointSampleWithDist | | √ | | | |
| FusedBiasLeakyrelu | | √ | | | | FusedBiasLeakyrelu | | √ | | | |
| GatherPoints | | √ | | | | GatherPoints | | √ | | | |
| GroupPoints | | √ | | | | GroupPoints | | √ | | | |
| Iou3d | | √ | √ | | | Iou3d | | √ | √ | | |
| KNN | | √ | | | | KNN | | √ | | | |
| MaskedConv | | √ | √ | | | MaskedConv | | √ | √ | | √ |
| MergeCells | | √ | | | | MergeCells | | √ | | | |
| MinAreaPolygon | | √ | | | | MinAreaPolygon | | √ | | | |
| ModulatedDeformConv2d | √ | √ | | | | ModulatedDeformConv2d | √ | √ | | | √ |
| MultiScaleDeformableAttn | | √ | √ | | | MultiScaleDeformableAttn | | √ | √ | | |
| NMS | √ | √ | √ | | | NMS | √ | √ | √ | | √ |
| NMSRotated | √ | √ | | | | NMSRotated | √ | √ | | | |
| NMSQuadri | √ | √ | | | | NMSQuadri | √ | √ | | | |
| PixelGroup | √ | | | | | PixelGroup | √ | | | | |
| PointsInBoxes | √ | √ | | | | PointsInBoxes | √ | √ | | | |
| PointsInPolygons | | √ | | | | PointsInPolygons | | √ | | | |
| PSAMask | √ | √ | √ | | | PSAMask | √ | √ | √ | | |
| RotatedFeatureAlign | √ | √ | | | | RotatedFeatureAlign | √ | √ | | | |
| RoIPointPool3d | | √ | √ | | | RoIPointPool3d | | √ | √ | | |
| RoIPool | | √ | √ | | | RoIPool | | √ | √ | | |
| RoIAlignRotated | √ | √ | √ | | | RoIAlignRotated | √ | √ | √ | | |
| RiRoIAlignRotated | | √ | | | | RiRoIAlignRotated | | √ | | | |
| RoIAlign | √ | √ | √ | | | RoIAlign | √ | √ | √ | | |
| RoIAwarePool3d | | √ | √ | | | RoIAwarePool3d | | √ | √ | | |
| SAConv2d | | √ | | | | SAConv2d | | √ | | | |
| SigmoidFocalLoss | | √ | √ | | | SigmoidFocalLoss | | √ | √ | | √ |
| SoftmaxFocalLoss | | √ | | | | SoftmaxFocalLoss | | √ | | | √ |
| SoftNMS | | √ | | | | SoftNMS | | √ | | | |
| Sparse Convolution | | √ | | | | Sparse Convolution | | √ | | | |
| Synchronized BatchNorm | | √ | | | | Synchronized BatchNorm | | √ | | | |
| ThreeInterpolate | | √ | | | | ThreeInterpolate | | √ | | | |
| ThreeNN | | √ | √ | | | ThreeNN | | √ | √ | | |
| TINShift | | √ | √ | | | TINShift | | √ | √ | | |
| UpFirDn2d | | √ | | | | UpFirDn2d | | √ | | | |
| Voxelization | √ | √ | | | | Voxelization | √ | √ | | | |
| PrRoIPool | | √ | | | | PrRoIPool | | √ | | | |

116
docs/zh_cn/understand_mmcv/ops.md
View File

@ -2,61 +2,61 @@
MMCV 提供了检测、分割等任务中常用的算子 MMCV 提供了检测、分割等任务中常用的算子
| Device | CPU | CUDA | MLU | MPS | | Device | CPU | CUDA | MLU | MPS | NPU |
| ---------------------------- | --- | ---- | --- | --- | | ---------------------------- | --- | ---- | --- | --- | --- |
| ActiveRotatedFilter | √ | √ | | | | ActiveRotatedFilter | √ | √ | | | |
| AssignScoreWithK | | √ | | | | AssignScoreWithK | | √ | | | |
| BallQuery | | √ | | | | BallQuery | | √ | | | |
| BBoxOverlaps | | √ | √ | √ | | BBoxOverlaps | | √ | √ | √ | |
| BorderAlign | | √ | | | | BorderAlign | | √ | | | |
| BoxIouRotated | √ | √ | | | | BoxIouRotated | √ | √ | | | |
| BoxIouQuadri | √ | √ | | | | BoxIouQuadri | √ | √ | | | |
| CARAFE | | √ | √ | | | CARAFE | | √ | √ | | |
| ChamferDistance | | √ | | | | ChamferDistance | | √ | | | |
| CrissCrossAttention | | √ | | | | CrissCrossAttention | | √ | | | |
| ContourExpand | √ | | | | | ContourExpand | √ | | | | |
| ConvexIoU | | √ | | | | ConvexIoU | | √ | | | |
| CornerPool | | √ | | | | CornerPool | | √ | | | |
| Correlation | | √ | | | | Correlation | | √ | | | |
| Deformable Convolution v1/v2 | √ | √ | | | | Deformable Convolution v1/v2 | √ | √ | | | |
| Deformable RoIPool | | √ | √ | | | Deformable RoIPool | | √ | √ | | |
| DiffIoURotated | | √ | | | | DiffIoURotated | | √ | | | |
| DynamicScatter | | √ | | | | DynamicScatter | | √ | | | |
| FurthestPointSample | | √ | | | | FurthestPointSample | | √ | | | |
| FurthestPointSampleWithDist | | √ | | | | FurthestPointSampleWithDist | | √ | | | |
| FusedBiasLeakyrelu | | √ | | | | FusedBiasLeakyrelu | | √ | | | |
| GatherPoints | | √ | | | | GatherPoints | | √ | | | |
| GroupPoints | | √ | | | | GroupPoints | | √ | | | |
| Iou3d | | √ | √ | | | Iou3d | | √ | √ | | |
| KNN | | √ | | | | KNN | | √ | | | |
| MaskedConv | | √ | √ | | | MaskedConv | | √ | √ | | √ |
| MergeCells | | √ | | | | MergeCells | | √ | | | |
| MinAreaPolygon | | √ | | | | MinAreaPolygon | | √ | | | |
| ModulatedDeformConv2d | √ | √ | | | | ModulatedDeformConv2d | √ | √ | | | √ |
| MultiScaleDeformableAttn | | √ | √ | | | MultiScaleDeformableAttn | | √ | √ | | |
| NMS | √ | √ | √ | | | NMS | √ | √ | √ | | √ |
| NMSRotated | √ | √ | | | | NMSRotated | √ | √ | | | |
| NMSQuadri | √ | √ | | | | NMSQuadri | √ | √ | | | |
| PixelGroup | √ | | | | | PixelGroup | √ | | | | |
| PointsInBoxes | √ | √ | | | | PointsInBoxes | √ | √ | | | |
| PointsInPolygons | | √ | | | | PointsInPolygons | | √ | | | |
| PSAMask | √ | √ | √ | | | PSAMask | √ | √ | √ | | |
| RotatedFeatureAlign | √ | √ | | | | RotatedFeatureAlign | √ | √ | | | |
| RoIPointPool3d | | √ | √ | | | RoIPointPool3d | | √ | √ | | |
| RoIPool | | √ | √ | | | RoIPool | | √ | √ | | |
| RoIAlignRotated | √ | √ | √ | | | RoIAlignRotated | √ | √ | √ | | |
| RiRoIAlignRotated | | √ | | | | RiRoIAlignRotated | | √ | | | |
| RoIAlign | √ | √ | √ | | | RoIAlign | √ | √ | √ | | |
| RoIAwarePool3d | | √ | √ | | | RoIAwarePool3d | | √ | √ | | |
| SAConv2d | | √ | | | | SAConv2d | | √ | | | |
| SigmoidFocalLoss | | √ | √ | | | SigmoidFocalLoss | | √ | √ | | √ |
| SoftmaxFocalLoss | | √ | | | | SoftmaxFocalLoss | | √ | | | √ |
| SoftNMS | | √ | | | | SoftNMS | | √ | | | |
| Sparse Convolution | | √ | | | | Sparse Convolution | | √ | | | |
| Synchronized BatchNorm | | √ | | | | Synchronized BatchNorm | | √ | | | |
| ThreeInterpolate | | √ | | | | ThreeInterpolate | | √ | | | |
| ThreeNN | | √ | √ | | | ThreeNN | | √ | √ | | |
| TINShift | | √ | √ | | | TINShift | | √ | √ | | |
| UpFirDn2d | | √ | | | | UpFirDn2d | | √ | | | |
| Voxelization | √ | √ | | | | Voxelization | √ | √ | | | |
| PrRoIPool | | √ | | | | PrRoIPool | | √ | | | |