下载
中文
注册

ScopeDecodeBboxPass

功能说明

将含有如下算子的Scope融合为DecodeBbox算子。Scope内包括:3的倍数个tf.Reshape算子、2的倍数个tf.Split算子、tf.Minimum算子、3的倍数个tf.Add算子 、tf. ConcatV2算子、2的倍数个tf.Sub算子,不包含Greater、Squeeze、Gather_2、TopKV2、boolean_mask等算子。

Scope详情

根据是否包含Transpose算子,有两种Scope,如下所示:

不含Transpose算子:

含Transpose算子:

融合后的算子原型

DecodeBbox,具体请参见算子加速库接口参考

融合对应关系

  • 不含Transpose算子:

    Reshape的输入作为融合后的输入box_predictions。

    Shape和Reshape_1的输入作为融合后的输入anchors。

    Reshape_2的输出为融合后的输出decoded_boxes。

    Reshape的输出作为Split的输入。

    Split的输出作为Minimum和Mul的输入。

    Minimum的输出作为Exp的输入。

    Exp的输出作为mul的输入。

    Reshape_1的输出作为split_1的输入。

    split_1的输出作为Sub和Add输入。

    Sub和Add的输出作为Mul的输入。

    Mul的输出作为Add_1、Sub_1、Add_2的输入。

    Add_1的输出作为Sub_1和Add_2的输入。

    Sub_1和Add_2的输出作为Concat的输入。

    Shape和Concat的输出作为Reshape_2的输入。

    融合算子排除Greater、Squeeze、Gather_2、TopKV2、boolean_mask等算子。

  • 含Transpose算子:

    transpose的输入作为融合后的输入box_predictions。

    transpose_1的输入作为融合后的输入anchors。

    transpose_2的输出为融合后的输出decoded_boxes。

    transpose的输出作为Reshape的输入。

    Reshape的输出作为Split的输入。

    Split的输出作为Minimum和Mul的输入。

    Minimum的输出作为Exp的输入。

    Exp的输出作为mul的输入。

    transpose_1的输出作为Reshape_1和Shape的输入。

    Reshape_1的输出作为split_1的输入。

    split_1的输出作为Sub和Add输入。

    Sub和Add的输出作为Mul的输入。

    Mul的输出作为Add_1、Sub_1、Add_2的输入。

    Add_1的输出作为Sub_1和Add_2的输入。

    Sub_1和Add_2的输出作为Concat的输入。

    Shape和Concat的输出作为Reshape_2的输入。

    Reshape_2的输出作为transpose_2的输入。

适用网络

2D-H1

融合规则类型

定制化融合规则