自定义算子导出ONNX支持清单

fast_gelu

1.8/1.11

npu_anchor_response_flags

1.8/1.11

npu_batch_nms

1.8/1.11

npu_bmmV2

1.8/1.11

npu_bounding_box_decode

1.8/1.11

npu_bounding_box_encode

1.8/1.11

npu_broadcast

1.8/1.11

npu_ciou

1.8/1.11

npu_confusion_transpose

1.8/1.11

npu_conv_transpose2d

1.8/1.11

npu_conv2d

1.8/1.11

npu_conv3d

1.8/1.11

npu_convolution

1.8/1.11

npu_convolution_transpose

1.8/1.11

npu_deformable_conv2d

1.8/1.11

npu_diou

1.8/1.11

npu_dropout_with_add_softmax

1.8/1.11

npu_dtype_cast

1.8/1.11

npu_fast_gelu

1.11

npu_format_cast

1.8/1.11

npu_fused_attention_layernorm_qkv_fwd

1.8/1.11

npu_fused_attention_score

1.8/1.11

npu_fused_attention_score_fwd

1.8/1.11

npu_giou

1.8/1.11

npu_grid_assign_positive

1.8/1.11

npu_gru

1.8/1.11

npu_ifmr

1.8/1.11

npu_indexing

1.8/1.11

npu_iou

1.8/1.11

npu_layer_norm_eval

1.11

npu_linear

1.8/1.11

npu_lstm

1.8/1.11

npu_lstm_cell

1.8/1.11

npu_masked_fill_range

1.8/1.11

npu_max

1.8/1.11

npu_min

1.8/1.11

npu_mish

1.8/1.11

npu_multi_head_attention

1.8/1.11

npu_nms_v4

1.8/1.11

npu_nms_with_mask

1.8/1.11

npu_normalize_batch

1.8/1.11

npu_one_hot

1.8/1.11

npu_pad

1.8/1.11

npu_ps_roi_pooling

1.8/1.11

npu_ptiou

1.8/1.11

npu_reshape

1.8/1.11

npu_roi_align

1.8/1.11

npu_rotated_box_decode

1.8/1.11

npu_rotated_box_encode

1.8/1.11

npu_rotated_iou

1.8/1.11

npu_rotated_overlaps

1.8/1.11

npu_scaled_masked_softmax

1.8/1.11

npu_scatter

1.8/1.11

npu_sign_bits_pack

1.8/1.11

npu_sign_bits_unpack

1.8/1.11

npu_silu

1.8/1.11

npu_slice

1.8/1.11

npu_softmax_cross_entropy_with_logits

1.8/1.11

npu_sort_v2

1.8/1.11

npu_stride_add

1.8/1.11

npu_stride_copy

1.8/1.11

npu_transpose

1.8/1.11

npu_yolo_boxes_encode

1.8/1.11

npu_nms_rotated

暂不支持