keras/src/ops/numpy.py/OnesLike/compute_output_spec
class OnesLike:    def compute_output_spec(self, x, dtype=None):
        if dtype is None:
            dtype = x.dtype
        return KerasTensor(x.shape, dtype=dtype)