code: def get_data(self, pp_index=None, tp_index=None):
      self._validate_indices(pp_index, tp_index)
      pp_indices = (
          list(range(self.pp_degree))
          if pp_index is None
          else [pp_index]
      )
      tp_indices = (
          list(range(self.tp_degree))
          if tp_index is None
          else [tp_index]
      )
  
      result = []
      for i in pp_indices:
          for j in tp_indices:
              result += self.map[self._make_key(i, j)]
  
      return result
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/checkpoint/reshape_meg_2d.py
imports: from .reshape_utils import partition_data
label_desc: Use image node_id 4 for calling the meg_2d_parallel_map obj's underlying member method code with example usage: obj.get_data(pp_index, tp_index) and returns: result