code: def compute_ReLU_flops(module, inp, out):
      assert isinstance(
          module, (nn.ReLU, nn.ReLU6, nn.PReLU, nn.ELU, nn.LeakyReLU)
      )
      batch_size = inp.size()[0]
      active_elements_count = batch_size
  
      for s in inp.size()[1:]:
          active_elements_count *= s
  
      return active_elements_count
  
file_path: tensorwatch/tensorwatch/model_graph/torchstat/compute_flops.py
imports: import torch.nn
  import torch
  import numpy
  import math
label_desc: Use image node_id 5 for calling a global function with example usage: compute_ReLU_flops(module, inp, out) and returns: active_elements_count