// FloatPerLayer-DAG: return %[[matmul1_uq]] : tensor<?x2xf32>
// FloatPerLayer-DAG: func.func private @composite_dot_general_with_bias_and_relu6_dynamic_fn_2
// FloatPerLayer-DAG: func.func private @composite_dot_general_with_bias_and_relu6_dynamic_fn_1

      (StableHLO_DynamicBroadcastInDimOp
        $bias,
        (Shape_ShapeOfOp $dot_general_1), $_, $_, $_)),
    (StableHLO_ConstantOp $cst_1)),
  (LiftAsTFXlaCallModule<"composite_dot_general_with_bias_and_relu6_dynamic_fn">
    (ArgumentList $lhs, $rhs, $bias),
    (ResultList $res),
    (NamedAttributeList
      (NamedAttr<"dot_dimension_numbers"> $dot_dimension_numbers),