code: def get_data_t(self, data, batch_size_t):
      data["texts"] = data["texts"][:batch_size_t]
      if "state" in data:
          h1 = data["state"]["td_hidden"][0][:batch_size_t]
          c1 = data["state"]["td_hidden"][1][:batch_size_t]
          h2 = data["state"]["lm_hidden"][0][:batch_size_t]
          c2 = data["state"]["lm_hidden"][1][:batch_size_t]
      else:
          h1, c1 = self.init_hidden_state(data["texts"])
          h2, c2 = self.init_hidden_state(data["texts"])
      data["state"] = {"td_hidden": (h1, c1), "lm_hidden": (h2, c2)}
      registry.register(f"{h1.device}_lstm_state", data["state"])
  
      return data, batch_size_t
  
file_path: mmf/tests/utils/test_model.py
imports: import torch
  from mmf.common.registry import registry
  from torch import nn
label_desc: Use image node_id 4 for calling the TestDecoderModel obj's underlying member method code with example usage: obj.get_data_t(data, batch_size_t) and returns: data,  batch_size_t