code: def inspect_keys_types(d: dict, prefix: Tuple = (), indent: int = 4):
      print_indent = lambda: print(" " * indent * len(prefix), end="")
      for k, v in d.items():
          if isinstance(v, dict):
              print_indent()
              print(f"> {k}:")
              inspect_keys_types(v, prefix + (k,), indent)
          else:
              print_indent()
              if isinstance(v, torch.Tensor):
                  print(f"> {k}: {type(v)} of shape {v.shape}")
              else:
                  print(f"> {k}: {type(v)}")
  
file_path: megatron-lm/megatron/core/dist_checkpointing/dict_utils.py
imports: from collections import defaultdict
  from typing import Any, Callable, Iterable, Optional, Tuple, Union
  import torch
label_desc: Use image node_id 3 for calling a global function with example usage: inspect_keys_types(d, prefix, indent) without return types