自定义算子导出ONNX支持清单
API名称 |
支持导出的PyTorch插件版本 |
|---|---|
fast_gelu |
2.1.0/2.3.1/2.4.0 |
npu_anchor_response_flags |
2.1.0/2.3.1/2.4.0 |
npu_batch_nms |
2.1.0/2.3.1/2.4.0 |
npu_bmmV2 |
2.1.0/2.3.1/2.4.0 |
npu_bounding_box_decode |
2.1.0/2.3.1/2.4.0 |
npu_bounding_box_encode |
2.1.0/2.3.1/2.4.0 |
npu_broadcast |
2.1.0/2.3.1/2.4.0 |
npu_ciou |
2.1.0/2.3.1/2.4.0 |
npu_confusion_transpose |
2.1.0/2.3.1/2.4.0 |
npu_conv_transpose2d |
2.1.0/2.3.1/2.4.0 |
npu_conv2d |
2.1.0/2.3.1/2.4.0 |
npu_conv3d |
2.1.0/2.3.1/2.4.0 |
npu_convolution |
2.1.0/2.3.1/2.4.0 |
npu_convolution_transpose |
2.1.0/2.3.1/2.4.0 |
npu_deformable_conv2d |
2.1.0/2.3.1/2.4.0 |
npu_diou |
2.1.0/2.3.1/2.4.0 |
npu_dropout_with_add_softmax |
2.1.0/2.3.1/2.4.0 |
npu_dtype_cast |
2.1.0/2.3.1/2.4.0 |
npu_format_cast |
2.1.0/2.3.1/2.4.0 |
npu_fused_attention_layernorm_qkv_fwd |
2.1.0/2.3.1/2.4.0 |
npu_giou |
2.1.0/2.3.1/2.4.0 |
npu_grid_assign_positive |
2.1.0/2.3.1/2.4.0 |
npu_gru |
2.1.0/2.3.1/2.4.0 |
npu_ifmr |
2.1.0/2.3.1/2.4.0 |
npu_indexing |
2.1.0/2.3.1/2.4.0 |
npu_iou |
2.1.0/2.3.1/2.4.0 |
npu_layer_norm_eval |
2.1.0/2.3.1/2.4.0 |
npu_linear |
2.1.0/2.3.1/2.4.0 |
npu_lstm |
2.1.0/2.3.1/2.4.0 |
npu_max |
2.1.0/2.3.1/2.4.0 |
npu_min |
2.1.0/2.3.1/2.4.0 |
npu_mish |
2.1.0/2.3.1/2.4.0 |
npu_nms_v4 |
2.1.0/2.3.1/2.4.0 |
npu_nms_with_mask |
2.1.0/2.3.1/2.4.0 |
npu_one_hot |
2.1.0/2.3.1/2.4.0 |
npu_pad |
2.1.0/2.3.1/2.4.0 |
npu_ps_roi_pooling |
2.1.0/2.3.1/2.4.0 |
npu_ptiou |
2.1.0/2.3.1/2.4.0 |
npu_reshape |
2.1.0/2.3.1/2.4.0 |
npu_roi_align |
2.1.0/2.3.1/2.4.0 |
npu_rotated_iou |
2.1.0/2.3.1/2.4.0 |
npu_rotated_overlaps |
2.1.0/2.3.1/2.4.0 |
npu_scatter |
2.1.0/2.3.1/2.4.0 |
npu_sign_bits_pack |
2.1.0/2.3.1/2.4.0 |
npu_sign_bits_unpack |
2.1.0/2.3.1/2.4.0 |
npu_silu |
2.1.0/2.3.1/2.4.0 |
npu_slice |
2.1.0/2.3.1/2.4.0 |
npu_softmax_cross_entropy_with_logits |
2.1.0/2.3.1/2.4.0 |
npu_sort_v2 |
2.1.0/2.3.1/2.4.0 |
npu_transpose |
2.1.0/2.3.1/2.4.0 |
npu_yolo_boxes_encode |
2.1.0/2.3.1/2.4.0 |