code: def _reshape_tp_dimension(old_2d_map, new_tp_degree):
      old_pp_degree = old_2d_map.pp_degree
      new_2d_map = meg_2d_parallel_map(old_pp_degree, new_tp_degree)
      for i in range(old_pp_degree):
          ranks_for_pp_index = old_2d_map.get_data(
              pp_index=i, tp_index=None
          )
          split_ranks = partition_data(
              ranks_for_pp_index, new_tp_degree
          )
          for j in range(new_tp_degree):
              new_2d_map.add_data(i, j, split_ranks[j])
  
      return new_2d_map
  
file_path: DeepSpeed/deepspeed/checkpoint/reshape_meg_2d.py
imports: from .reshape_utils import partition_data
label_desc: Use image node_id 1 for calling a global function with example usage: _reshape_tp_dimension(old_2d_map, new_tp_degree) and returns: new_2d_map