Skip to content

Commit

Permalink
Update resources.h
Browse files Browse the repository at this point in the history
  • Loading branch information
yifeif authored Apr 23, 2018
1 parent a202f54 commit 829ec05
Showing 1 changed file with 0 additions and 20 deletions.
20 changes: 0 additions & 20 deletions tensorflow/core/kernels/boosted_trees/resources.h
Original file line number Diff line number Diff line change
Expand Up @@ -82,26 +82,6 @@ class BoostedTreesEnsembleResource : public StampedResource {

int64 GetNumNodes(const int32 tree_id);

void UpdateLastLayerNodesRange(const int32 node_range_start,
int32 node_range_end) const {
tree_ensemble_->mutable_growing_metadata()->set_last_layer_node_start(
node_range_start);
tree_ensemble_->mutable_growing_metadata()->set_last_layer_node_end(
node_range_end);
}

void GetLastLayerNodesRange(int32* node_range_start,
int32* node_range_end) const {
*node_range_start =
tree_ensemble_->growing_metadata().last_layer_node_start();
*node_range_end = tree_ensemble_->growing_metadata().last_layer_node_end();
}

int64 GetNumNodes(const int32 tree_id) {
DCHECK_LT(tree_id, tree_ensemble_->trees_size());
return tree_ensemble_->trees(tree_id).nodes_size();
}

void UpdateGrowingMetadata() const;

int32 GetNumLayersAttempted();
Expand Down

0 comments on commit 829ec05

Please sign in to comment.