code: def right_shift(x1: ArrayLike, x2: ArrayLike, /) -> Array:
      x1, x2 = promote_args_numeric(np.right_shift.__name__, x1, x2)
      lax_fn = (
          lax.shift_right_logical
          if np.issubdtype(x1.dtype, np.unsignedinteger)
          else lax.shift_right_arithmetic
      )
      return lax_fn(x1, x2)
  
imports: from __future__ import annotations
  from functools import partial
  import operator
  from textwrap import dedent
  from typing import Any, Callable, overload
  import numpy
  from jax._src import core
  from jax._src import dtypes
  from jax._src.api import jit
  from jax._src.custom_derivatives import custom_jvp
  from jax._src.lax import lax
  from jax._src.typing import Array, ArrayLike
  from jax._src.numpy.util import check_arraylike, promote_args, promote_args_inexact, promote_args_numeric, promote_dtypes_inexact, promote_dtypes_numeric, promote_shapes, _where, _wraps, check_no_float0s
label_desc: Use image node_id 13 for calling a global function with example usage: right_shift() and returns: lax_fn