// CHECK: "tf.XlaCallModule"(%[[dq_act]], %[[dq_weight]]
    %1 = "tf.XlaCallModule"(%0, %cst) {
      Sout = [#tf_type.shape<2x2>], config = "",
      _entry_function = @composite_dot_general,
      module = "composite_dot_general",
      platforms = [], version = 4 : i64
    } : (tensor<2x2xf32>, tensor<2x2xf32>) -> tensor<2x2xf32>

        # function name starts with "composite_dot_general".
        specs=qc.QuantizationSpecs(
            specs=[
                qc.QuantizationSpec(
                    matcher=qc.MatcherSpec(
                        function_name=qc.FunctionNameMatcherSpec(
                            regex='composite_dot_general.*'
                        )
                    ),

// quantizable units.
constexpr absl::string_view kSpecsDisableAllDotGeneral =
    R"pb(specs
         [ {
           matcher { function_name { regex: "composite_dot_general_.*" } }
           method { no_quantization {} }
         }])pb";

// Configure `QuantizationSpecs` to apply `StaticRangePtq` to all quantizable
// units.
constexpr absl::string_view kSpecsStaticRangePtqToAll =