Searched refs:increment_num_updates (Results 1 – 2 of 2) sorted by relevance
146 const int32 increment_num_updates) { in SetTreeWeight() argument150 if (increment_num_updates != 0) { in SetTreeWeight()154 ->set_num_tree_weight_updates(num_updates + increment_num_updates); in SetTreeWeight()
597 std::vector<int32> increment_num_updates(num_trees, 0); in UpdateTreeWeightsIfDropout() local600 1 /* only 1 tree was added */, &weights, &increment_num_updates); in UpdateTreeWeightsIfDropout()605 increment_num_updates[i]); in UpdateTreeWeightsIfDropout()