Searched refs:new_num_layers (Results 1 – 4 of 4) sorted by relevance
106 const int32 new_num_layers = in Compute() local108 VLOG(1) << "Adding layer #" << new_num_layers - 1 << " to tree #" in Compute()154 ensemble_resource->SetNumLayersGrown(current_tree, new_num_layers); in Compute()
78 void SetNumLayersGrown(const int32 tree_id, int32 new_num_layers) const;
133 const int32 tree_id, int32 new_num_layers) const { in SetNumLayersGrown()136 new_num_layers); in SetNumLayersGrown()
634 const auto new_num_layers = tree_metadata->num_layers_grown() + 1; in UpdateAndRetrieveGrowableTree() local635 VLOG(1) << "Adding layer #" << new_num_layers - 1 << " to tree #" in UpdateAndRetrieveGrowableTree()638 tree_metadata->set_num_layers_grown(new_num_layers); in UpdateAndRetrieveGrowableTree()639 tree_metadata->set_is_finalized(new_num_layers >= max_tree_depth); in UpdateAndRetrieveGrowableTree()