def lexicographic_comparator(*tf_args: TfVal) -> TfVal:
    return _convert_jax_impl(lax._sort_lt_comparator, multiple_results=False)(*tf_args, _in_avals=comparator_jax_in_avals, _out_aval=core.ShapedArray((), np.bool_), num_keys=num_keys)
def lexicographic_comparator(*tf_args: TfVal) -> TfVal:
    return _convert_jax_impl(lax._sort_lt_comparator, multiple_results=False)(*tf_args, _in_avals=comparator_jax_in_avals, _out_aval=core.ShapedArray((), np.bool_), num_keys=num_keys)