func.func private @quantize_simple_xla_call_module_no_operand() -> tensor<1x3xf32> {

  EXPECT_THAT(xla_call_module_op,
              HasSymNameAttr("_entry_function",
                             StartsWith("composite_dot_general_fn")));
  EXPECT_THAT(xla_call_module_op,
              HasStringAttr("_original_entry_function",
                            StartsWith("composite_dot_general_fn")));

  // Tests that there are CustomAggregatorOps inserted.
  SmallVector<TF::CustomAggregatorOp> custom_aggregator_ops;

      %1 = "tf.XlaCallModule"(%0, %cst) {Sout = [#tf_type.shape<1x3>], _entry_function = @composite_dot_general_fn, _original_entry_function = "composite_dot_general_fn", _quantization_method = "static_range_ptq { }", _stablehlo_module_attrs = {}, _tfl_quant_trait = "fully_quantizable", device = "", dim_args_spec = [], disabled_checks = [], has_token_input_output = false, module...

    return %2 : tensor<1x3xf32>
  }
  // CHECK-LABEL: func.func private @composite_dot_general_fn_1
  // CHECK-SAME: -> tensor<1x3xf32>

    return %0 : tensor<1x3xf32>
    // CHECK: %[[CALL:.+]] = "tf.XlaCallModule"(%[[ARG0]], %[[ARG1]])
    // CHECK-SAME: _entry_function = @composite_dot_general_fn_1
    // CHECK-SAME: _original_entry_function = "composite_dot_general_fn_1"
    // CHECK: return %[[CALL]]
  }

  // CHECK: @composite_dot_general_fn_1
  // CHECK-SAME: %[[ARG2:[^:[:space:]]+]]
  // CHECK-SAME: %[[ARG3:[^:[:space:]]+]]

// quantization method, implying "no_quantization".
// EMPTY-SAME: _entry_function = @composite_dot_general_fn_1
// EMPTY-SAME: _original_entry_function
// EMPTY-NOT: _quantization_method
// EMPTY-SAME: _tfl_quant_trait = "fully_quantizable"

// EMPTY: return %[[XLA_CALL_MODULE:.+]] : tensor<1x1x64xf32>
// EMPTY: }

// EMPTY-LABEL: private @composite_dot_general_fn_1
// EMPTY-SAME: tf_quant.composite_function

    %2 = "tf.XlaCallModule"(%1) <{Sout = [#tf_type.shape<3x10>],...

  return %2 : tensor<1x3xf32>
}

// CHECK: %[[XLA_CALL_MODULE:.+]] = "tf.XlaCallModule"(%[[CUSTOM_AGGREGATOR_0]], %[[CST]])
// CHECK-SAME: _entry_function = @composite_dot_general_fn_1, _original_entry_function = "composite_dot_general_fn_1"

    return %1 : tensor<1x3xf32>
  }