def __init__(self, params, device_assigner, training=True, tree_variables_class=TreeTrainingVariables):
    self.variables = []
    for i in range(params.num_trees):
        with ops.device(device_assigner.get_device(i)):
            self.variables.append(tree_variables_class(params, i, training))
def __init__(self, params, device_assigner, training=True, tree_variables_class=TreeTrainingVariables):
    self.variables = []
    for i in range(params.num_trees):
        with ops.device(device_assigner.get_device(i)):
            self.variables.append(tree_variables_class(params, i, training))