code: def find_modules(nn_module, type):
      return [
          module
          for module in nn_module.modules()
          if isinstance(module, type)
      ]
  
file_path: vit-pytorch/vit_pytorch/recorder.py
imports: from functools import wraps
  import torch
  from torch import nn
  from vit_pytorch.vit import Attention
label_desc: Use image node_id 1 for calling a global function with example usage: find_modules(nn_module, type) and returns: unknown