code: def __init__(self, hidden_size, eps=1e-5, memory_efficient=False):
      super().__init__()
      self.epsilon = eps
      self.memory_efficient = memory_efficient
      self.weight = torch.nn.Parameter(torch.empty(hidden_size))
      self.bias = torch.nn.Parameter(torch.empty(hidden_size))
      self.reset_parameters()
  
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 to create a new FastLayerNorm object from inherited base classes: torch.nn with example: obj = FastLayerNorm(hidden_size, eps, memory_efficient)