code: def _piecewise_constant(boundaries, values, t):
      index = jnp.sum(boundaries < t)
      return jnp.take(values, index)
  
imports: import jax.numpy
  import numpy
  from absl import logging
label_desc: Use image node_id 1 for calling a global function with example usage: _piecewise_constant(boundaries, values, t) and returns: jnp