// The input module is expected to be an MHLO module, or a quantized StableHLO
// graph (expressed as `mlir::TF::XlaCallModuleOp`s) if
// `pass_config.enable_stablehlo_quantizer` is `true`.
void AddPostQuantizationStableHloToTfPasses(
    const mlir::TFL::PassConfig& pass_config,
    mlir::OpPassManager& pass_manager);

// This is the early part of the conversion in isolation. This enables a caller

    // StableHLO because the quantizer takes StableHLO dialect as its input.
    pass_manager.addPass(mlir::mhlo::createHloLegalizeToStablehloPass());
  }
}

void AddPostQuantizationStableHloToTfPasses(
    const mlir::TFL::PassConfig& pass_config,
    mlir::OpPassManager& pass_manager) {
  if (pass_config.enable_stablehlo_quantizer) {

      LOG(ERROR) << "Failed to run quantization: "
                 << quantized_module_op.status();
      return mlir::failure();
    }
    module = *quantized_module_op;
  }

  AddPostQuantizationStableHloToTfPasses(pass_config, pass_manager);
  if (failed(pass_manager.run(module))) {
    return mlir::failure();
  }
  pass_manager.clear();

  return mlir::success();
}

}  // namespace