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