class_docstr: Float 16 Module.
  
  Attributes:
      config (TransformerConfig): Transformer config
      fp16 (bool) : Specifies if the model runs in fp16 mode
      bf16 (bool) : Specifies if the model runs in bf16 mode
  
  Args:
      config (TransformerConfig): The transformer config used to initalize the model
code: def forward(self, *inputs, **kwargs):
      if parallel_state.is_pipeline_first_stage():
          inputs = fp32_to_float16(inputs, self.float16_convertor)
      outputs = self.module(*inputs, **kwargs)
      if parallel_state.is_pipeline_last_stage():
          outputs = float16_to_fp32(outputs)
      return outputs
  
imports: import torch
  from torch.autograd import Variable
  from torch.nn.parameter import Parameter
  from megatron.core import parallel_state
  from megatron.core.transformer.transformer_config import TransformerConfig
label_desc: Use image node_id 3 for calling the Float16Module obj's underlying member method code with example usage: obj.forward() and returns: outputs