}

// Quantizable Unit matcher that uses lifted function's name for matching.
class FunctionNameMatcher {
 public:
  explicit FunctionNameMatcher(const FunctionNameMatcherSpec& spec)
      : match_regex_(GetMatchRegex(spec)) {}

  // Returns `true` when matched with the entry function of
  // `xla_call_module_op`.
  bool Match(TF::XlaCallModuleOp xla_call_module_op) const {
    if (match_regex_ == nullptr) return false;