code: def tie_in_original_fn(f, init_fn, apply_fn):
      # EXPERIMENTAL: Expose the original function as a private attribute.
      if isinstance(f, (Transformed, TransformedWithState)):
          f = getattr(f.init, "_original_fn")
      init_fn._original_fn = f  # pylint: disable=protected-access
      apply_fn._original_fn = f
  
imports: from collections.abc import Mapping
  import inspect
  from typing import Any, Callable, NamedTuple, Optional, TypeVar, Union
  from haiku._src import analytics
  from haiku._src import base
  from haiku._src import data_structures
  from haiku._src import typing
  import jax
label_desc: Use image node_id 10 for calling a global function with example usage: tie_in_original_fn(f, init_fn, apply_fn) without return types