code: def _givens_rotation(a, b):
      b_zero = abs(b) == 0
      a_lt_b = abs(a) < abs(b)
      t = -jnp.where(a_lt_b, a, b) / jnp.where(a_lt_b, b, a)
      r = lax.rsqrt(1 + abs(t) ** 2).astype(t.dtype)
      cs = jnp.where(b_zero, 1, jnp.where(a_lt_b, r * t, r))
      sn = jnp.where(b_zero, 0, jnp.where(a_lt_b, r, r * t))
      return cs, sn
  
imports: from functools import partial
  import operator
  import numpy
  import jax
  import jax.numpy
  from jax import device_put
  from jax import lax
  from jax import scipy
  from jax.tree_util import tree_leaves, tree_map, tree_structure, tree_reduce, Partial
  from jax._src import dtypes
  from jax._src.lax import lax
  from jax._src.util import safe_map
label_desc: Use image node_id 19 for calling a global function with example usage: _givens_rotation(a, b) and returns: cs,  sn