diff --git a/docs/en/understand_mmcv/ops.md b/docs/en/understand_mmcv/ops.md index e0a9a3648..8a4bfe88e 100644 --- a/docs/en/understand_mmcv/ops.md +++ b/docs/en/understand_mmcv/ops.md @@ -2,61 +2,61 @@ We implement common ops used in detection, segmentation, etc. -| Device | CPU | CUDA | MLU | MPS | -| ---------------------------- | --- | ---- | --- | --- | -| ActiveRotatedFilter | √ | √ | | | -| AssignScoreWithK | | √ | | | -| BallQuery | | √ | | | -| BBoxOverlaps | | √ | √ | √ | -| BorderAlign | | √ | | | -| BoxIouRotated | √ | √ | | | -| BoxIouQuadri | √ | √ | | | -| CARAFE | | √ | √ | | -| ChamferDistance | | √ | | | -| CrissCrossAttention | | √ | | | -| ContourExpand | √ | | | | -| ConvexIoU | | √ | | | -| CornerPool | | √ | | | -| Correlation | | √ | | | -| Deformable Convolution v1/v2 | √ | √ | | | -| Deformable RoIPool | | √ | √ | | -| DiffIoURotated | | √ | | | -| DynamicScatter | | √ | | | -| FurthestPointSample | | √ | | | -| FurthestPointSampleWithDist | | √ | | | -| FusedBiasLeakyrelu | | √ | | | -| GatherPoints | | √ | | | -| GroupPoints | | √ | | | -| Iou3d | | √ | √ | | -| KNN | | √ | | | -| MaskedConv | | √ | √ | | -| MergeCells | | √ | | | -| MinAreaPolygon | | √ | | | -| ModulatedDeformConv2d | √ | √ | | | -| MultiScaleDeformableAttn | | √ | √ | | -| NMS | √ | √ | √ | | -| NMSRotated | √ | √ | | | -| NMSQuadri | √ | √ | | | -| 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 | | √ | | | +| Device | CPU | CUDA | MLU | MPS | NPU | +| ---------------------------- | --- | ---- | --- | --- | --- | +| ActiveRotatedFilter | √ | √ | | | | +| AssignScoreWithK | | √ | | | | +| BallQuery | | √ | | | | +| BBoxOverlaps | | √ | √ | √ | | +| BorderAlign | | √ | | | | +| BoxIouRotated | √ | √ | | | | +| BoxIouQuadri | √ | √ | | | | +| CARAFE | | √ | √ | | | +| ChamferDistance | | √ | | | | +| CrissCrossAttention | | √ | | | | +| ContourExpand | √ | | | | | +| ConvexIoU | | √ | | | | +| CornerPool | | √ | | | | +| Correlation | | √ | | | | +| Deformable Convolution v1/v2 | √ | √ | | | | +| Deformable RoIPool | | √ | √ | | | +| DiffIoURotated | | √ | | | | +| DynamicScatter | | √ | | | | +| FurthestPointSample | | √ | | | | +| FurthestPointSampleWithDist | | √ | | | | +| FusedBiasLeakyrelu | | √ | | | | +| GatherPoints | | √ | | | | +| GroupPoints | | √ | | | | +| Iou3d | | √ | √ | | | +| KNN | | √ | | | | +| MaskedConv | | √ | √ | | √ | +| MergeCells | | √ | | | | +| MinAreaPolygon | | √ | | | | +| ModulatedDeformConv2d | √ | √ | | | √ | +| MultiScaleDeformableAttn | | √ | √ | | | +| NMS | √ | √ | √ | | √ | +| NMSRotated | √ | √ | | | | +| NMSQuadri | √ | √ | | | | +| 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 | | √ | | | | diff --git a/docs/zh_cn/understand_mmcv/ops.md b/docs/zh_cn/understand_mmcv/ops.md index 6b4622146..7cf062ef7 100644 --- a/docs/zh_cn/understand_mmcv/ops.md +++ b/docs/zh_cn/understand_mmcv/ops.md @@ -2,61 +2,61 @@ MMCV 提供了检测、分割等任务中常用的算子 -| Device | CPU | CUDA | MLU | MPS | -| ---------------------------- | --- | ---- | --- | --- | -| ActiveRotatedFilter | √ | √ | | | -| AssignScoreWithK | | √ | | | -| BallQuery | | √ | | | -| BBoxOverlaps | | √ | √ | √ | -| BorderAlign | | √ | | | -| BoxIouRotated | √ | √ | | | -| BoxIouQuadri | √ | √ | | | -| CARAFE | | √ | √ | | -| ChamferDistance | | √ | | | -| CrissCrossAttention | | √ | | | -| ContourExpand | √ | | | | -| ConvexIoU | | √ | | | -| CornerPool | | √ | | | -| Correlation | | √ | | | -| Deformable Convolution v1/v2 | √ | √ | | | -| Deformable RoIPool | | √ | √ | | -| DiffIoURotated | | √ | | | -| DynamicScatter | | √ | | | -| FurthestPointSample | | √ | | | -| FurthestPointSampleWithDist | | √ | | | -| FusedBiasLeakyrelu | | √ | | | -| GatherPoints | | √ | | | -| GroupPoints | | √ | | | -| Iou3d | | √ | √ | | -| KNN | | √ | | | -| MaskedConv | | √ | √ | | -| MergeCells | | √ | | | -| MinAreaPolygon | | √ | | | -| ModulatedDeformConv2d | √ | √ | | | -| MultiScaleDeformableAttn | | √ | √ | | -| NMS | √ | √ | √ | | -| NMSRotated | √ | √ | | | -| NMSQuadri | √ | √ | | | -| 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 | | √ | | | +| Device | CPU | CUDA | MLU | MPS | NPU | +| ---------------------------- | --- | ---- | --- | --- | --- | +| ActiveRotatedFilter | √ | √ | | | | +| AssignScoreWithK | | √ | | | | +| BallQuery | | √ | | | | +| BBoxOverlaps | | √ | √ | √ | | +| BorderAlign | | √ | | | | +| BoxIouRotated | √ | √ | | | | +| BoxIouQuadri | √ | √ | | | | +| CARAFE | | √ | √ | | | +| ChamferDistance | | √ | | | | +| CrissCrossAttention | | √ | | | | +| ContourExpand | √ | | | | | +| ConvexIoU | | √ | | | | +| CornerPool | | √ | | | | +| Correlation | | √ | | | | +| Deformable Convolution v1/v2 | √ | √ | | | | +| Deformable RoIPool | | √ | √ | | | +| DiffIoURotated | | √ | | | | +| DynamicScatter | | √ | | | | +| FurthestPointSample | | √ | | | | +| FurthestPointSampleWithDist | | √ | | | | +| FusedBiasLeakyrelu | | √ | | | | +| GatherPoints | | √ | | | | +| GroupPoints | | √ | | | | +| Iou3d | | √ | √ | | | +| KNN | | √ | | | | +| MaskedConv | | √ | √ | | √ | +| MergeCells | | √ | | | | +| MinAreaPolygon | | √ | | | | +| ModulatedDeformConv2d | √ | √ | | | √ | +| MultiScaleDeformableAttn | | √ | √ | | | +| NMS | √ | √ | √ | | √ | +| NMSRotated | √ | √ | | | | +| NMSQuadri | √ | √ | | | | +| 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 | | √ | | | |