code: def compute_Linear_flops(module, inp, out):
      assert isinstance(module, nn.Linear)
      assert len(inp.size()) == 2 and len(out.size()) == 2
      batch_size = inp.size()[0]
      return batch_size * inp.size()[1] * out.size()[1]
  
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 7 for calling a global function with example usage: compute_Linear_flops(module, inp, out) and returns: unknown