docstr: Partition an input tensor along the last dimension, aligned with given splits.
  
  Subclasses should override this method to account for new input types.
  
  Args:
      input (List[Tensor]): The tensor to partition along the last dimension.
      split_sizes (List[int]): The size of each partition.
  
  Returns:
      List[Any]: A list of the chunks of ``input``.
docstr_tok: ["Partition","an","input","tensor","along","the","last","dimension",",","aligned","with","given","splits",".","Subclasses","should","override","this","method","to","account","for","new","input","types",".","Args",":","input","(","List","[","Tensor","]",")",":","The","tensor","to","partition","along","the","last","dimension",".","split_sizes","(","List","[","int","]",")",":","The","size","of","each","partition",".","Returns",":","List","[","Any","]",":","A","list","of","the","chunks","of","``","input","``","."]
code: def _split_global_input(self, input, split_sizes):
      """Partition an input tensor along the last dimension, aligned with given splits.
  
      Subclasses should override this method to account for new input types.
  
      Args:
          input (List[Tensor]): The tensor to partition along the last dimension.
          split_sizes (List[int]): The size of each partition.
  
      Returns:
          List[Any]: A list of the chunks of ``input``.
      """
      return split_tensor_along_last_dim(input, split_sizes)
  
file_path: DeepSpeed/deepspeed/runtime/zero/tiling.py
imports: import torch
  import deepspeed
  from deepspeed.runtime.utils import partition_uniform
label_desc: Use image node_id 3 for calling the TiledLinear obj's underlying member method code with example usage: obj._split_global_input(input, split_sizes) and returns: split_tensor_along_last_dim