Home
last modified time | relevance | path

Searched refs:new_num_layers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc97 const int32 new_num_layers = in Compute() local
99 VLOG(1) << "Adding layer #" << new_num_layers - 1 << " to tree #" in Compute()
127 ensemble_resource->SetNumLayersGrown(current_tree, new_num_layers); in Compute()
303 const int32 new_num_layers = in Compute() local
305 VLOG(1) << "Adding layer #" << new_num_layers - 1 << " to tree #" in Compute()
347 ensemble_resource->SetNumLayersGrown(current_tree, new_num_layers); in Compute()
Dresources.h80 void SetNumLayersGrown(const int32 tree_id, int32 new_num_layers) const;
Dresources.cc138 const int32 tree_id, int32 new_num_layers) const { in SetNumLayersGrown()
141 new_num_layers); in SetNumLayersGrown()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6424 int new_num_layers = converter->network()->getNbLayers(); in ConvertGraphDefToEngine() local
6425 for (int i = num_layers; i < new_num_layers; i++) { in ConvertGraphDefToEngine()
6437 num_layers = new_num_layers; in ConvertGraphDefToEngine()