ModuleOp module, absl::string_view checkpoint_path);

// Creates a pass that sinks SavedModel asset filenames to constants.
std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateAssetSinkingPass(
    llvm::StringRef saved_model_dir);

}  // namespace tf_saved_model
}  // namespace mlir

  module->setAttr("tf_saved_model.semantics",
                  UnitAttr::get(builder.getContext()));
  return absl::OkStatus();
}

std::unique_ptr<OperationPass<ModuleOp>> CreateAssetSinkingPass(
    llvm::StringRef saved_model_dir) {
  return std::make_unique<AssetSinkingPass>(saved_model_dir);
}

}  // namespace tf_saved_model