code: def __init__(
      self,
      normalized_shape,
      eps: float = 1e-5,
      elementwise_affine: bool = True,
      *,
      sequence_parallel_enabled: bool = False,
  ):
      super().__init__(
          normalized_shape=normalized_shape,
          eps=eps,
          elementwise_affine=elementwise_affine,
      )
      self.sequence_parallel_enabled = sequence_parallel_enabled
      if self.elementwise_affine:
          _set_sequence_parallel_enabled(
              self.weight, self.sequence_parallel_enabled
          )
          _set_sequence_parallel_enabled(
              self.bias, self.sequence_parallel_enabled
          )
  
file_path: apex/apex/transformer/layers/layer_norm.py
imports: import warnings
  import torch
  from apex.normalization import FusedLayerNorm
  from apex.normalization import MixedFusedLayerNorm
label_desc: Use image node_id 1 to create a new FusedLayerNorm object from inherited base classes: OrigFusedLayerNorm with example: obj = FusedLayerNorm(normalized_shape, eps, elementwise_affine)