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