class_docstr: all tables assigned to this class instance are managed by a single CachedEmbeddingBag.
  Those parameters in TablewiseEmbeddingBagConfig are ignored: cuda_row_num, buffer_size, initial_weight.
class_docstr_tok: ["all","tables","assigned","to","this","class","instance","are","managed","by","a","single","CachedEmbeddingBag",".","Those","parameters","in","TablewiseEmbeddingBagConfig","are","ignored",":","cuda_row_num",",","buffer_size",",","initial_weight","."]
code: def element_size(self):
      return self.weight.element_size()
  
file_path: colossalai/colossalai/legacy/nn/parallel/layers/cache_embedding/parallel_cached_embedding_tablewise.py
imports: from typing import List
  import torch
  import torch.distributed
  import torch.nn.functional
  from colossalai.legacy.nn._ops._utils import dual_all_to_all_tablewise
  from colossalai.legacy.tensor import ProcessGroup
  from .cache_mgr import EvictionStrategy
  from .cached_embedding import CachedEmbeddingBag
  from .embedding_config import TablewiseEmbeddingBagConfig
label_desc: Use image node_id 6 for calling the ParallelCachedEmbeddingBagTablewise obj's underlying member method code with example usage: obj.element_size() and returns: self