code: def simple_init(self):
      self.map = {
          self._make_key(i // self.tp_degree, i % self.tp_degree): [i]
          for i in range(self.pp_degree * self.tp_degree)
      }
  
file_path: deepspeed/deepspeed/checkpoint/reshape_meg_2d.py
imports: from .reshape_utils import partition_data
label_desc: Use image node_id 2 for calling the meg_2d_parallel_map obj's underlying member method code with example usage: obj.simple_init() without return types