code: def compute_ReLU_memory(module, inp, out):
      assert isinstance(
          module, (nn.ReLU, nn.ReLU6, nn.ELU, nn.LeakyReLU)
      )
  
      mread = inp.numel()
      mwrite = out.numel()
  
      return mread * inp.element_size(), mwrite * out.element_size()
  
file_path: tensorwatch/tensorwatch/model_graph/torchstat/compute_memory.py
imports: import torch.nn
  import torch
  import numpy
label_desc: Use image node_id 3 for calling a global function with example usage: compute_ReLU_memory(module, inp, out) and returns: unknown,  unknown