def get_stats(self, session):
    tree_stats = []
    for i in range(self.params.num_trees):
        with ops.device(self.device_assigner.get_device(i)):
            tree_stats.append(self.trees[i].get_stats(session))
    return ForestStats(tree_stats, self.params)
def get_stats(self, session):
    tree_stats = []
    for i in range(self.params.num_trees):
        with ops.device(self.device_assigner.get_device(i)):
            tree_stats.append(self.trees[i].get_stats(session))
    return ForestStats(tree_stats, self.params)