src/transformers/models/transfo_xl/modeling_tf_transfo_xl.py/TFTransfoXLLMHeadModel/reset_memory_length
class TFTransfoXLLMHeadModel:    def reset_memory_length(self, mem_len):
        self.transformer.reset_memory_length(mem_len)