keras/src/layers/preprocessing/discretization.py/Discretization/compute_output_spec
class Discretization:    def compute_output_spec(self, inputs):
        return backend.KerasTensor(shape=inputs.shape, dtype=self.compute_dtype)