支持算子

    附加算子

    OP NameHostX86CUDAARMOpenCLFPGA华为NPU百度XPU瑞芯微NPU联发科APU
    abs　　YY　　　　　　
    anchor_generator　　　Y　　　　　　
    assignY　　　　　　　　　
    attention_padding_mask　　　　　　　　　　
    axpy　　　Y　　　　　　
    beam_search_decode　　　Y　　　　　　
    beam_search_decode　　　Y　　　　　　
    box_clip　　　Y　　　　　　
    calib_once　　YY　Y　　　　
    clip　　　Y　　　　　　
    collect_fpn_proposals　　　Y　　　　　　
    conditional_blockY　　　　　　　　　
    crf_decodingY　　　　　　　　　
    crop　　　Y　　　　　　
    ctc_alignY　　　　　　　　　
    decode_bboxes　　　Y　　　　　　
    deformable_conv　　　Y　　　　　　
    distribute_fpn_proposals　　　Y　　　　　　
    equalY　　　　　　　　　
    exp　　　YY　　　　　
    fake_channel_wise_dequantize_max_abs　　　　　　　　　　
    fake_dequantize_max_abs　　　　　　　　　　
    fake_quantize_abs_max　　　　　　　　　　
    fake_quantize_dequantize_abs_max　　　　　　　　　　
    fake_quantize_dequantize_moving_average_abs_max　　　　　　　　　　
    fake_quantize_moving_average_abs_max　　　　　　　　　　
    fake_quantize_range_abs_max　　　　　　　　　　
    floor　　　Y　　　　　　
    gather　Y　Y　　　Y　　
    gelu　Y　　　　　　　　
    generate_proposals　　　Y　　　　　　
    greater_equalY　　　　　　　　　
    greater_thanY　　　　　　　　　
    group_norm　　　Y　　　　　　
    gru　YYY　Y　　　　
    gru_unit　　　Y　　　　　　
    hard_sigmoid　　　YY　Y　　　
    hard_swish　　　Y　　　　　　
    im2sequence　　　Y　　　　　　
    increment　　　Y　　Y　　　
    is_emptyY　　　　　　　　　
    layer_norm　Y　Y　　YY　　
    layout_once　　YY　Y　　　　
    less_equalY　　　　　　　　　
    less_thanY　　　　　Y　　　
    lod_reset　　　Y　　　　　　
    log　　　Y　　Y　　　
    logical_andY　　　　　　　　　
    logical_notY　　　　　　　　　
    logical_orY　　　　　　　　　
    logical_xorY　　　　　　　　　
    lookup_table　YYY　　　Y　　
    lookup_table_dequant　　　Y　　　　　　
    lookup_table_v2　YYY　　　　　　
    lrn　　　YY　　　　　
    lstm　　　Y　　　　　　
    match_matrix_tensor　YY　　　　　　　
    max_pool2d_with_index　　　　　　　　　　
    mean　　　Y　　　　　　
    merge_lod_tensor　　　Y　　　　　　
    negative　　　Y　　　　　　
    norm　　　Y　Y　　　　
    not_equalY　　　　　　　　　
    one_hotY　　　　　　　　　
    pixel_shuffleY　　YY　　　　　
    pow　　　Y　　　　　　
    printY　　　　　　　　　
    read_from_arrayY　　　　　　　　　
    reciprocal　　　Y　　　　　　
    reduce_max　　　Y　　　　　　
    reduce_prod　　　Y　　　　　　
    reduce_sum　Y　　　　　Y　　
    relu_clipped　　　Y　　Y　　　
    retinanet_detection_outputY　　　　　　　　　
    roi_align　　　Y　　　　　　
    rsqrt　　　Y　　　　　　
    search_aligned_mat_mul　YY　　　　　　　
    search_attention_padding_mask　YY　　　　　　　
    search_grnn　YY　　　　　　　
    search_group_padding　YY　　　　　　　
    search_seq_arithmetic　YY　　　　　　　
    search_seq_depadding　YY　　　　　　　
    search_seq_fc　YY　　　　　　　
    search_seq_softmax　YY　　　　　　　
    sequence_arithmetic　YY　　　　　　　
    sequence_concat　YY　　　　　　　
    sequence_conv　Y　Y　　　　　　
    sequence_expand　　　Y　　　　　　
    sequence_expand_as　Y　　　　　　　　
    sequence_mask　　Y　　　　　　　
    sequence_pad　　Y　　　　　　　
    sequence_pool　YYY　　　　　　
    sequence_pool_concat　　Y　　　　　　　
    sequence_reshape　Y　　　　　　　　
    sequence_reverse　YY　　　　　　　
    sequence_reverse_embedding　　Y　　　　　　　
    sequence_softmax　　　Y　　　　　　
    sequence_unpadY　Y　　　　　　　
    shapeYY　　　　　　　　
    sign　　　　　　　　　　
    softsign　Y　　　　Y　　　
    split_lod_tensor　　　Y　　　　　　
    sqrt　　　　　　Y　　　
    square　Y　Y　　Y　　　
    swish　　　YY　　　　　
    top_k　　　Y　　　　　　
    topk_pooling　　Y　　　　　　　
    uniform_random　　　　　　　　　　
    var_conv_2d　YY　　　　　　　
    where_indexY　　　　　　　　　
    whileY　　　　　　　　　
    write_to_arrayY　　　　　　　　　
    xpuconv2d　　　　　　　Y　　
    xpuembedding_with_eltwise_add　　　　　　　Y　　
    xpufc　　　　　　　Y　　
    xpummdnn_bid_emb_att　　　　　　　Y　　
    xpummdnn_bid_emb_grnn_att　　　　　　　Y　　
    xpummdnn_bid_emb_grnn_att2　　　　　　　Y　　
    xpummdnn_match_conv_topk　　　　　　　Y　　
    xpummdnn_merge_all　　　　　　　Y　　
    xpummdnn_search_attention　　　　　　　Y　　
    xpumulti_encoder　　　　　　　Y　　
    xpuresnet_cbam　　　　　　　Y　　
    xpuresnet50　　　　　　　Y　　
    xpusfa_head　　　　　　　Y　　
    matrix_nmsY