def __init__(self, tensor_access_list, tensors):
    self.tensor_access_list = tensor_access_list
    self.tensors = tensors
    self.next_swap_tasks_index = 0
def __init__(self, tensor_access_list, tensors):
    self.tensor_access_list = tensor_access_list
    self.tensors = tensors
    self.next_swap_tasks_index = 0