code: def compute_ReLU_madd(module, inp, out):
      assert isinstance(module, (nn.ReLU, nn.ReLU6))
  
      count = 1
      for i in inp.size()[1:]:
          count *= i
      return count
  
file_path: tensorwatch/tensorwatch/model_graph/torchstat/compute_madd.py
imports: import torch.nn
label_desc: Use image node_id 6 for calling a global function with example usage: compute_ReLU_madd(module, inp, out) and returns: count