code: def _fast_layer_norm(x, weight, bias, epsilon, memory_efficient):
      args = _cast_if_autocast_enabled(
          x, weight, bias, epsilon, memory_efficient
      )
      with torch.cuda.amp.autocast(enabled=False):
          return FastLayerNormFN.apply(*args)
  
file_path: apex/apex/contrib/layer_norm/layer_norm.py
imports: import torch
  from torch.nn import init
  from apex._autocast_utils import _cast_if_autocast_enabled
  import fast_layer_norm
label_desc: Use image node_id 1 for calling a global function with example usage: _fast_layer_norm(x, weight, bias, epsilon, memory_efficient) and returns: FastLayerNormFN