keras/src/ops/numpy.py/SearchSorted/compute_output_spec
class SearchSorted:    def compute_output_spec(self, sorted_sequence, values, side="left"):
        if len(sorted_sequence.shape) != 1:
            raise ValueError(
                "searchsorted only supports 1-D sorted sequences. Use"
                "keras.ops.vectorized_map to extend to N-D sequences."
            )
        out_type = (
            "int32"
            if sorted_sequence.shape[0] <= np.iinfo(np.int32).max
            else "int64"
        )
        return KerasTensor(values.shape, dtype=out_type)