Searched refs:leaf_examples (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
| D | stats_ops.cc | 203 const std::unordered_map<int32, std::vector<int>>& leaf_examples, in UpdateStatsCollated() argument
206 auto it = leaf_examples.begin(); in UpdateStatsCollated()
208 auto end_it = leaf_examples.begin(); in UpdateStatsCollated()
277 std::unordered_map<int32, std::vector<int>> leaf_examples; in Compute() local
280 leaf_examples[leaf_ids(i)].push_back(i); in Compute()
292 const int32 num_leaves = leaf_examples.size(); in Compute()
321 tree_resource, &leaf_examples, &set_lock, in Compute()
327 target, num_targets, leaf_examples, &set_lock, in Compute()
|