def _tbptt_split_batch(self, batch: Any) -> List[Any]:
    splits = [batch]
    truncated_bptt_enabled = self._truncated_bptt_enabled()
    if truncated_bptt_enabled:
        model_ref = self.trainer.lightning_module
        with self.trainer.profiler.profile('tbptt_split_batch'):
            splits = model_ref.tbptt_split_batch(batch, self._truncated_bptt_steps())
    return splits
def _tbptt_split_batch(self, batch: Any) -> List[Any]:
    splits = [batch]
    truncated_bptt_enabled = self._truncated_bptt_enabled()
    if truncated_bptt_enabled:
        model_ref = self.trainer.lightning_module
        with self.trainer.profiler.profile('tbptt_split_batch'):
            splits = model_ref.tbptt_split_batch(batch, self._truncated_bptt_steps())
    return splits