code: def _register_orig_params(
      state: _FSDPState, module: nn.Module
  ) -> None:
      """
      Deregisters the ``FlatParameter``; registers the original parameters.
      """
      handle = _module_handle(state, module)
      if not handle:
          return
      _deregister_flat_param(state, module)
      if handle.is_sharded(handle.flat_param):
          handle._use_sharded_views()
          handle._use_sharded_grad_views()
      else:
          handle._use_unsharded_views(as_params=True)
  
file_path: pytorch/torch/distributed/fsdp/_unshard_param_utils.py
imports: import contextlib
  import warnings
  from typing import cast, Generator
  import torch
  import torch.distributed.fsdp._traversal_utils
  import torch.nn
  from torch.distributed.fsdp._common_utils import _FSDPState, _has_fsdp_params, _module_handle, HandleTrainingState, TrainingState
  from torch.distributed.fsdp._runtime_utils import _get_fsdp_root_states_with_modules, _lazy_init, _reset_flat_param_grad_info_if_needed, _reshard, _reshard_grads, _unshard, _unshard_grads
  from torch.distributed.utils import _p_assert
  from ._flat_param import FlatParamHandle
label_desc: Use image node_id 10 for calling a global function with example usage: _register_orig_params(state, module) without return types