def args_maker():
    'Test the set of inputs onp.geomspace is well-defined on.'
    (start, stop) = self._GetArgsMaker(rng, [start_shape, stop_shape], [dtype, dtype])()
    (start, stop) = jnp.broadcast_arrays(start, stop)
    if (dtype in complex_dtypes):
        return (start, stop)
    start = ((start * jnp.sign(start)) * jnp.sign(stop))
    return (start, stop)
Test the set of inputs onp.geomspace is well-defined on.
def args_maker():
    
    (start, stop) = self._GetArgsMaker(rng, [start_shape, stop_shape], [dtype, dtype])()
    (start, stop) = jnp.broadcast_arrays(start, stop)
    if (dtype in complex_dtypes):
        return (start, stop)
    start = ((start * jnp.sign(start)) * jnp.sign(stop))
    return (start, stop)