def METHOD_NAME(self, inputs, training=True):
    inputs = self.backend.cast(inputs, self.compute_dtype)
    if training:
        return self._randomly_adjust_brightness(inputs)
    else:
        return inputs