code: def __init__(
      self,
      normalized_shape,
      eps: float = 1e-5,
      **kwargs,
  ) -> None:
      self.sequence_parallel_enabled = kwargs.get(
          "sequence_parallel_enabled", False
      )
      super().__init__(
          normalized_shape=normalized_shape, eps=eps, **kwargs
      )
      if self.sequence_parallel_enabled:
          _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 MixedFusedLayerNorm object from inherited base classes: OrigMixedFusedLayerNorm with example: obj = MixedFusedLayerNorm(normalized_shape, eps)