keras/src/layers/preprocessing/text_vectorization.py/TextVectorization/update_state
class TextVectorization:    def update_state(self, data):
        self._lookup_layer.update_state(self._preprocess(data))