code: def addbmm_impl(input, batch1, batch2, beta=1, alpha=1):
      if alpha != 1 and beta != 1:
          return (
              torch.bmm(batch1, batch2.transpose(1, 2)) * alpha
              + input * beta
          )
      elif alpha != 1:
          return (
              torch.bmm(batch1, batch2.transpose(1, 2)) * alpha + input
          )
      elif beta != 1:
          return (
              torch.bmm(batch1, batch2.transpose(1, 2)) + input * beta
          )
      else:
          return torch.bmm(batch1, batch2.transpose(1, 2)) + input
  
file_path: ColossalAI/colossalai/_analyzer/fx/tracer/bias_addition.py
imports: import torch
  import torch.nn.functional
  from torch.nn.modules.utils import _pair, _single, _triple
  from .tracer import register_tracer_impl
label_desc: Use image node_id 9 for calling a global function with example usage: addbmm_impl(input, batch1, batch2, beta, alpha) and returns: unknown, unknown, unknown, unknown