// This is the early part of the conversion in isolation. This enables a caller
// to inject more information in the middle of the conversion before resuming it
// (like freezing variables for example).
void AddPreVariableFreezingTFToTFLConversionPasses(
    const mlir::TFL::PassConfig& pass_config,
    mlir::OpPassManager* pass_manager);

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

}

// This is the early part of the conversion in isolation. This enables a caller
// to inject more information in the middle of the conversion before resuming
// it.
void AddPreVariableFreezingTFToTFLConversionPasses(
    const mlir::TFL::PassConfig& pass_config,
    mlir::OpPassManager* pass_manager) {
  // This pass wraps all the tf.FakeQuant ops in a custom op so they are not

            quantization_py_function_lib, saved_model_bundle, pass_manager,
            status_handler, module))) {
      return status_handler.ConsumeStatus();
    }
  }

  AddPreVariableFreezingTFToTFLConversionPasses(pass_config, &pass_manager);
  if (failed(pass_manager.run(module))) {
    return status_handler.ConsumeStatus();
  }

  // Freeze variables if a session is provided.
  if (session != nullptr &&