code: def _vprim_abstract_eval(
      *inputs, prim, __axis_size, __axis_name, __batch_axes, params
  ):
      assert len(inputs) == len(__batch_axes)
      inputs = [
          jax.core.mapped_aval(__axis_size, b, x)
          for x, b in zip(inputs, __batch_axes)
      ]
      abstract_eval = _vprim_abstract_eval_registry[prim]
      outs = abstract_eval(*inputs, **params)
      outs = [
          jax.core.unmapped_aval(__axis_size, __axis_name, 0, x)
          for x in outs
      ]
      return outs
  
imports: import functools
  from typing import Any
  import jax
  import jax.core
  import jax.interpreters.ad
  import jax.interpreters.batching
  import jax.interpreters.mlir
  import jax.tree_util
  from jaxtyping import PyTree
  from .._filters import combine, is_array, is_array_like, partition
  from .._tree import tree_equal
label_desc: Use image node_id 18 for calling a global function with example usage: _vprim_abstract_eval() and returns: outs