//        |         |
    //      Const_0    Const_1
    //
    const float const0_val[] = {1.0, 2.0, 3.0, 4.0};
    const float const1_val[] = {1.0, 0.0, 0.0, 1.0};
    TF_Operation* const0 = FloatConst2x2(graph_, s_, const0_val, "Const_0");
    TF_Operation* const1 = FloatConst2x2(graph_, s_, const1_val, "Const_1");
    TF_Operation* matmul = MatMul(graph_, s_, const0, const1, "MatMul");
    inputs[0] = TF_Output{const0, 0};

def UndoBroadcastConvBiasAdd : Pat<
  (TFL_AddOp $lhs, (Arith_ConstantOp:$const_op $bias), TFL_AF_None),
  (TFL_AddOp $lhs, (Arith_ConstantOp (FlattenTo1D $bias)), TFL_AF_None),
  [(AnyStaticShapeTensor $lhs),
   (IsLastDimEqualToNumElements $bias, $bias),
   (HasOneUse $const_op),
   (HasRankAtMost<4> $bias),
   (HasRankAtLeast<2> $bias),
   (IsDefinedByConv2DOp $lhs)]>;