code: def _apply_functions_after_first_forward(
      module: torch.nn.Module,
      # pyre-ignore[2]
      input: Any,
      # pyre-ignore[2]
      output: Any,
  ) -> None:
      _functions_to_lazy_apply = getattr(
          module, "_functions_to_lazy_apply", None
      )
      if _functions_to_lazy_apply is not None:
          for fn in _functions_to_lazy_apply:
              module.apply(fn)
          delattr(module, "_functions_to_lazy_apply")
      module._lazy_apply_hook.remove()
      delattr(module, "_lazy_apply_hook")
  
file_path: torchrec/torchrec/modules/lazy_extension.py
imports: import functools
  import inspect
  from typing import Any, Callable
  import torch
  import torch.utils.hooks
  from torch.nn.modules.lazy import _LazyProtocol, LazyModuleMixin
  from torch.nn.modules.module import _global_backward_hooks, _global_forward_hooks, _global_forward_pre_hooks
label_desc: Use image node_id 1 for calling a global function with example usage: _apply_functions_after_first_forward(module, input, output) without return types