code: def _ref_raise_to_shaped(ref_aval: AbstractRef, weak_type):
      return AbstractRef(
          core.raise_to_shaped(ref_aval.inner_aval, weak_type)
      )
  
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 1 for calling a global function with example usage: _ref_raise_to_shaped(ref_aval, weak_type) and returns: AbstractRef