code: def set_full_hp_param(self, value, optim_state_key=None):
      if self._hp_mapping is not None:
          lp_frag_address = self._hp_mapping.lp_fragment_address
          value_fragment = torch.narrow(
              value.flatten(),
              0,
              lp_frag_address.start,
              lp_frag_address.numel,
          )
          hp_fragment = self._hp_mapping.get_hp_fragment(
              optim_state_key
          )
          hp_fragment.data.copy_(value_fragment.data)
  
imports: import torch
  from dataclasses import dataclass
  from deepspeed import comm
  from typing import Dict
label_desc: Use image node_id 2 for calling a global function with example usage: set_full_hp_param(value, optim_state_key) without return types