code: def get_optim_state_keys(self):
      return list(self.optim_fragment.keys())
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/deepspeed/utils/tensor_fragment.py
imports: import torch
  from dataclasses import dataclass
  from deepspeed import comm
label_desc: Use image node_id 5 for calling the tensor_fragment obj's underlying member method code with example usage: obj.get_optim_state_keys() and returns: list