code: def deepspeed_param_size(p: torch.nn.Parameter) -> int:
      assert hasattr(p, "ds_numel")
      return p.ds_numel
  
file_path: pytorch-lightning/src/lightning/pytorch/utilities/model_summary/model_summary_deepspeed.py
imports: from collections import OrderedDict
  from typing import Dict, List, Tuple
  import torch
  from lightning_utilities.core.imports import RequirementCache
  from torch.nn import Parameter
  from lightning.pytorch.utilities.model_summary.model_summary import NOT_APPLICABLE, LayerSummary, ModelSummary, _is_lazy_weight_tensor, get_human_readable_count
label_desc: Use image node_id 1 for calling a global function with example usage: deepspeed_param_size(p) and returns: p