(self, a, v, side="left")
| 1649 | return jnp.argsort(a, axis) |
| 1650 | |
| 1651 | def searchsorted(self, a, v, side="left"): |
| 1652 | if a.ndim == 1: |
| 1653 | return jnp.searchsorted(a, v, side) |
| 1654 | else: |
| 1655 | return jax.vmap(lambda b, u: jnp.searchsorted(b, u, side))(a, v) |
| 1656 | |
| 1657 | def flip(self, a, axis=None): |
| 1658 | return jnp.flip(a, axis) |
nothing calls this directly
no test coverage detected