#include "tensorflow/compiler/mlir/lite/utils/convert_type.h"
#include "tensorflow/compiler/mlir/lite/utils/utils.h"

//===----------------------------------------------------------------------===//
// The ReduceTypePrecision Pass.
//
namespace mlir {
namespace TFL {

namespace {

#define GEN_PASS_DEF_REDUCETYPEPRECISIONPASS
#include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc"

            << pass_config.enable_stablehlo_conversion
            << "\nlegalize_custom_tensor_list_ops: "
            << pass_config.legalize_custom_tensor_list_ops
            << "\nreduce_type_precision: " << pass_config.reduce_type_precision
            << "\nconvert_qdq_format: "
            << GetQDQQuantModeString(pass_config.qdq_conversion_mode) << "\n";
}

}  // namespace TFL
}  // namespace mlir

  pass_config.enable_hlo_to_tf_conversion = enable_hlo_to_tf_conversion;
  pass_config.disable_hlo_to_tfl_conversion = disable_hlo_to_tfl_conversion;
  pass_config.reduce_type_precision = reduce_type_precision;

  toco::TocoFlags toco_flags;
  toco_flags.set_force_select_tf_ops(!emit_builtin_tflite_ops);
  toco_flags.set_enable_select_tf_ops(emit_select_tf_ops);

extern llvm::cl::opt<bool> enable_dynamic_update_slice;
extern llvm::cl::opt<bool> preserve_assert_op;
extern llvm::cl::opt<bool> legalize_custom_tensor_list_ops;
extern llvm::cl::opt<bool> reduce_type_precision;

// Import saved model.
extern llvm::cl::opt<bool> import_saved_model_object_graph;
extern llvm::cl::opt<bool> import_saved_model_signature_defs;
extern llvm::cl::opt<std::string> saved_model_tags;

      AddDynamicRangeQuantizationPasses(pass_config, *pass_manager);
    }
    pass_manager->addPass(mlir::createCanonicalizerPass());

    if (pass_config.reduce_type_precision ||
        toco_flags.reduce_type_precision()) {
      pass_manager->addPass(mlir::TFL::CreateReduceTypePrecisionPass());
    }

    // This pass should be always at the end of the model

// NOLINTNEXTLINE
opt<bool> serialize_stablehlo_ops(
    "serialize-stablehlo-ops",
    llvm::cl::desc("Wether serialize stablehlo ops or not"),
    llvm::cl::init(true));

// NOLINTNEXTLINE
opt<bool> reduce_type_precision(
    "reduce-type-precision",
    llvm::cl::desc("Convert tensors to a lower precision if all values are "
                   "within the reduced precision range. This could have side "

        "transforms/pin_ops_with_side_effects.cc",
        "transforms/prepare_composite_functions_tf.cc",
        "transforms/prepare_tf.cc",
        "transforms/raise_custom_ops.cc",
        "transforms/reduce_type_precision.cc",
        "transforms/reduce_while_operands.cc",
        "transforms/runtime_verify.cc",
        "transforms/split_merged_operands.cc",
        "transforms/trim_functions_tf.cc",