code: def fused_rms_norm_affine(
      input, weight, normalized_shape, eps=1e-6, memory_efficient=False
  ):
      args = _cast_if_autocast_enabled(
          input, weight, normalized_shape, eps, memory_efficient
      )
      with torch.cuda.amp.autocast(enabled=False):
          return FusedRMSNormAffineFunction.apply(*args)
  
imports: import importlib
  import numbers
  import torch
  from torch.nn.parameter import Parameter
  from torch.nn import init
  from torch.nn import functional
  from apex._autocast_utils import _cast_if_autocast_enabled
label_desc: Use image node_id 5 for calling a global function with example usage: fused_rms_norm_affine(input, weight, normalized_shape, eps, memory_efficient) and returns: FusedRMSNormAffineFunction