code: def _addmm_flops_compute(
      input, mat1, mat2, *, beta=1, alpha=1, out=None
  ):
      """
      Count flops for the addmm operation.
      """
      macs = _prod(mat1.shape) * mat2.shape[-1]
      return 2 * macs + _prod(input.shape), 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 22 for calling a global function with example usage: _addmm_flops_compute(input, mat1, mat2) and returns: unknown,  macs