code: def find_relu_layers(model, layer_type):
      modules = []
  
      for _, module in model.named_modules():
          if isinstance(module, layer_type):
              modules.append(module)
  
      return modules
  
imports: import inspect
  import pytest
  import torch
  import torch.nn
  import torch.nn.functional
  import torchvision.models
  from flashtorch.saliency import Backprop
label_desc: Use image node_id 2 for calling a global function with example usage: find_relu_layers(model, layer_type) and returns: modules