code: def forward(self, x):
      return rms_norm(x, self.weight, self.eps)
  
file_path: flash-attention/flash_attn/ops/rms_norm.py
imports: import torch
  from torch.nn import init
  from flash_attn.ops.layer_norm import DropoutAddLayerNormFn, DropoutAddLayerNormParallelResidualFn, DropoutAddLayerNormSubsetFn
label_desc: Use image node_id 3 for calling the RMSNorm obj's underlying member method code with example usage: obj.forward(x) and returns: rms_norm