&bundle);
  } else if (import_hlo) {
    // HLO import path.
    std::string error;
    std::unique_ptr<llvm::MemoryBuffer> buffer =
        mlir::openInputFile(input_file_name, &error);
    if (buffer == nullptr) {
      llvm::errs() << "Cannot open input file: " << input_file_name << " "
                   << error;
      return kTrFailure;
    }

  mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context);
  {
    mlir::Location loc = mlir::UnknownLoc::get(&context);
    std::string error;
    std::unique_ptr<llvm::MemoryBuffer> buffer =
        mlir::openInputFile(flatbuffer_file, &error);
    if (buffer == nullptr) {
      TF_SetStatus(status, TF_INVALID_ARGUMENT,
                   ("Unable to load input file " + error).c_str());
      return;
    }

    absl::string_view control_output_arrays, llvm::SourceMgr* source_mgr,
    MLIRContext* context) {
  // Set up the input file.
  std::string error_message;
  auto file = mlir::openInputFile(input_filename, &error_message);
  if (!file) {
    return absl::InvalidArgumentError(
        absl::StrCat("Failed to open input file: ", error_message));
  }

  if (input_mlir) {