Searched refs:ready_to_split (Results 1 – 1 of 1) sorted by relevance
154 std::unordered_set<int32>* ready_to_split) { in UpdateStats() argument191 ready_to_split->insert(leaf_id); in UpdateStats()205 std::unordered_set<int32>* ready_to_split) { in UpdateStatsCollated() argument217 ready_to_split->insert(leaf_id); in UpdateStatsCollated()301 std::unordered_set<int32> ready_to_split; in Compute() local311 fertile_stats_resource, &locks, &set_lock, &ready_to_split, in Compute()317 static_cast<int32>(end), &ready_to_split); in Compute()322 &ready_to_split, &data_set, in Compute()329 &ready_to_split); in Compute()342 output_shape.AddDim(ready_to_split.size()); in Compute()[all …]