8755c48bb903005a7e819998d51ba41ba892814f,src/graph_builder/frontend/sub_rules/concat_affine.py,ConcatAffine,_compress_sequence,#ConcatAffine#Any#,104

Before Change

        const_bias = ConstantVariable(merged_bias, OrderC)
        bias_op = AxiswiseBias(conv_op.name + "_tail_bias", axis=Axis.C)
        bias_op.append_input("x", conv_out)
        bias_op.append_input("b", const_bias)
        bias_op.append_output("y", final_out)

After Change

        conv_weight_var.data *= merged_scale[broadcast]

        final_out = seq[-1].outputs["y"]
        if bias_found:
            // Scale/Biasレイヤーを削除して、新しいBiasレイヤーを元々の出力につなぐ
            for op in seq[1:]:
                op.remove_all()
            const_bias = ConstantVariable(merged_bias, OrderC)
            bias_op = AxiswiseBias(conv_op.name + "_tail_bias", axis=Axis.C)
            bias_op.append_input("x", conv_out)
            bias_op.append_input("b", const_bias)
            bias_op.append_output("y", final_out)
        else:
            // Biasはないので、Convレイヤーの出力がブロック全体の出力になる
            for op in seq[1:]:
                op.remove_all()
            conv_op.remove_output(conv_out)
            conv_op.append_output("y", final_out)

Instances