code: def _matmul_flops_compute(input, other, *, out=None):
      """
      Count flops for the matmul operation.
      """
      macs = _prod(input.shape) * other.shape[-1]
      return 2 * macs, macs
  
imports: import time
  import torch
  import torch.nn
  import torch.nn.functional
  from functools import partial
  from typing import List, Optional
  from collections import OrderedDict
  import numpy
label_desc: Use image node_id 21 for calling a global function with example usage: _matmul_flops_compute(input, other) and returns: unknown,  macs