class_docstr: High-level wrapper for GPU embedding cache
code: def __init__(self, num_items, num_feats, idtype=F.int64):
      assert idtype in [F.int32, F.int64]
      self._cache = _CAPI_DGLGpuCacheCreate(
          num_items, num_feats, 32 if idtype == F.int32 else 64
      )
      self.idtype = idtype
      self.total_miss = 0
      self.total_queries = 0
  
file_path: dgl/python/dgl/cuda/gpu_cache.py
imports: from ..None import backend
  from .._ffi.function import _init_api
label_desc: Use image node_id 1 to create a new GPUCache object from inherited base classes: object with example: obj = GPUCache(num_items, num_feats, idtype)