code: def _repeat_abstract_eval(x, *, repeats, axis):
      shape = list(x.shape)
      shape[axis] *= repeats
      return jax_core.ShapedArray(shape, x.dtype)
  
imports: from __future__ import annotations
  import contextlib
  import dataclasses
  import enum
  from typing import Any, Callable
  import jax
  from jax._src import api_util
  from jax._src import core
  from jax._src import effects
  from jax._src import linear_util
  from jax._src import pretty_printer
  from jax._src import state
  from jax._src import tree_util
  from jax._src import util
  from jax._src.state import indexing
  from jax._src.state import primitives
  from jax._src.interpreters import mlir
  from jax._src.interpreters import partial_eval
  from jax._src.pallas.mosaic import core
  import jax.numpy
label_desc: Use image node_id 2 for calling a global function with example usage: _repeat_abstract_eval(x) and returns: jax_core