code: def _unmap_ref(size, axis_name, axis, ref_aval):
      return AbstractRef(
          core.unmapped_aval(size, axis_name, axis, ref_aval.inner_aval)
      )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import dataclasses
  import math
  from typing import Any, Generic, TypeVar, Union
  from jax._src import core
  from jax._src import effects
  from jax._src import pretty_printer
  from jax._src.state import indexing
  from jax._src.util import safe_map, safe_zip
  from jax._src.typing import Array
label_desc: Use image node_id 3 for calling a global function with example usage: _unmap_ref(size, axis_name, axis, ref_aval) and returns: AbstractRef