class_docstr: Bert Language model.
docstr: See megatron.model.transformer.set_input_tensor()
docstr_tok: ["See","megatron.model.transformer.set_input_tensor","(",")"]
code: def set_input_tensor(self, input_tensor):
      """See megatron.model.transformer.set_input_tensor()"""
      self.language_model.set_input_tensor(input_tensor)
  
imports: import contextlib
  import torch
  from apex.transformer import tensor_parallel
  from apex.transformer.enums import AttnMaskType
  from apex.transformer.enums import ModelType
  from apex.transformer.layers import FusedLayerNorm
  from apex.transformer.testing.global_vars import get_args
  from apex.transformer.testing.standalone_transformer_lm import MegatronModule, get_language_model, get_linear_layer, init_method_normal, scaled_init_method_normal, parallel_lm_logits
label_desc: Use image node_id 2 for calling the BertModel obj's underlying member method code with example usage: obj.set_input_tensor(input_tensor) without return types