}  // namespace TF
}  // namespace mlir

namespace llvm {
template <>
struct DenseMapInfo<mlir::TF::ResourceHandle> {
  static mlir::TF::ResourceHandle getEmptyKey() {
    return {/*container=*/"", /*name=*/"", /*device=*/"",
            /*op=*/DenseMapInfo<mlir::Operation*>::getEmptyKey()};
  }

  static mlir::TF::ResourceHandle getTombstoneKey() {

  std::optional<GpuDeviceMetadata> GetGpuDeviceMetadata(
      const ParsedName& device) const;

 private:
  llvm::SmallVector<ParsedName, 8> device_names_;
  // TODO(ezhulenev): Add DenseMapInfo<ParsedName> specialization to be able to
  // use ParsedName as a key in a DenseMap.
  llvm::StringMap<GpuDeviceMetadata> gpu_metadata_;
};

}  // namespace TF
}  // namespace mlir

  }

  KernelSpecs& WithImpl(const ScaleDecomposeFn& dfn) {
    decompose_fn_ = dfn;
    return *this;
  }

 private:
  // The signature is pattern match based.
  struct SignatureInfo : public llvm::DenseMapInfo<Signature> {
    static inline Signature getEmptyKey() { return {}; }
    static inline Signature getTombstoneKey() { return {nullptr}; }
    static unsigned getHashValue(Signature val) {