code: def layernorm_cx(cx, w_in):
      """Accumulates complexity of layernorm into cx = (h, w, flops, params, acts)."""
      h, w, flops, params, acts = (
          cx["h"],
          cx["w"],
          cx["flops"],
          cx["params"],
          cx["acts"],
      )
      params += 2 * w_in
      return {
          "h": h,
          "w": w,
          "flops": flops,
          "params": params,
          "acts": acts,
      }
  
imports: import numpy
  import torch
  import torch.nn
  from pycls.core.config import cfg
  from torch.nn import Module
label_desc: Use image node_id 14 for calling a global function with example usage: layernorm_cx(cx, w_in) and returns: dict