code: def elem_size(dtype: torch.dtype) -> int:
      """
      Return size in bytes of the given dtype.
      """
      try:
          return ELEM_SIZES[dtype]
      except KeyError:
          raise ValueError("Unknown dtype size for {}".format(dtype))
  
file_path: deepspeed/deepspeed/inference/v2/inference_utils.py
imports: from typing import Dict
  import torch
  from enum import Enum, IntEnum
label_desc: Use image node_id 2 for calling a global function with example usage: elem_size(dtype) and returns: ELEM_SIZES