class SaveQuantizationReportInstrumentation : public PassInstrumentation {
 public:
  // `file_path` is the path to save the report file. The report file is in
  // textproto format so a `.txtpb` extension is preferred but it doesn't result
  // in error if other extension is used. This instrumentation will not be run
  // if `file_path` is a `nullopt`.
  explicit SaveQuantizationReportInstrumentation(

  QuantizeCompositeFunctionsPassOptions options;
  pm.addPass(createQuantizeCompositeFunctionsPass(options));

  const std::string report_file_path =
      absl::StrCat(testing::TempDir(), "/save_report.txtpb");
  pm.addInstrumentation(std::make_unique<SaveQuantizationReportInstrumentation>(
      report_file_path));

  const LogicalResult run_result = pm.run(*module_op);
  ASSERT_TRUE(succeeded(run_result));

    path_map = {'serving_default': dataset_path}
    repr_dataset.TfRecordRepresentativeDatasetSaver(path_map).save(
        {'serving_default': data_gen()}
    )

    report_file_path = self.create_tempfile('report.txtpb').full_path
    config = qc.QuantizationConfig(
        static_range_ptq_preset=qc.StaticRangePtqPreset(
            representative_datasets=[
                qc.RepresentativeDatasetConfig(

      ParseModuleOpString(kQuantizedDotGeneral);
  ASSERT_TRUE(module_op);

  const QuantizationReport report(*module_op);

  const std::string dst_file_path =
      absl::StrCat(TempDir(), "/quantization_report.txtpb");
  const absl::Status save_status = report.Save(dst_file_path);
  ASSERT_THAT(save_status, IsOk());

  const absl::StatusOr<std::string> file_data = ReadFileToString(dst_file_path);
  ASSERT_THAT(file_data, IsOk());

// Tests that `composite_dot_general_fn_1` and its corresponding XlaCallModuleOp
// contains attributes required for quantization, including the
// `_quantization_method` attribute that contains textpb of `Method`.

// DISABLE-ALL-DOT-GENERAL: @main
func.func @main(%arg0: tensor<1x1x167xf32>) -> tensor<1x1x64xf32> {
  %0 = stablehlo.constant dense<2.000000e+00> : tensor<167x64xf32>