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 set_input_tensor(self, input_tensor):
      return self.module.set_input_tensor(input_tensor)
  
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 2 for calling the Float16Module obj's underlying member method code with example usage: obj.set_input_tensor(input_tensor) and returns: self