public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(CheckAcceptedOpsPass)

  explicit CheckAcceptedOpsPass(
      const std::vector<std::string> &optional_accepted_dialects)
      : accepted_dialects_(GetAcceptedStableHLODialects()),
        optional_accepted_dialects_(optional_accepted_dialects) {}

  // Check if TF dialect ops exist over the module.
  void runOnOperation() override;

 private:

#include <string>
#include <vector>

#include "llvm/ADT/StringRef.h"

namespace mlir {
namespace odml {

std::vector<std::string> GetAcceptedStableHLODialects();

std::vector<std::string> GetAcceptedTFLiteDialects();

// Can we find the given `dialect_name` in the `accepted_dialects`?
bool IsAcceptedDialect(llvm::StringRef dialect_name,

#include <algorithm>
#include <string>
#include <vector>

#include "llvm/ADT/DenseSet.h"

namespace mlir {
namespace odml {

std::vector<std::string> GetAcceptedStableHLODialects() {
  // It returns the default list of accepted dialects.
  std::vector<std::string> accepted_dialects({"stablehlo", "builtin", "func"});
  return accepted_dialects;
}