Searched refs:LevelSize (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/lib/random/
H A Dweighted_picker.cc32 while (LevelSize(num_levels_ - 1) < N) {
39 level_[l] = new int32[LevelSize(l)];
149 for (int i = N_; i < LevelSize(num_levels_ - 1); i++) leaves[i] = 0;
161 for (int i = N_; i < LevelSize(num_levels_ - 1); i++) leaves[i] = 0;
171 for (int i = 0; i < LevelSize(l); i++) {
184 if (new_size <= LevelSize(num_levels_ - 1)) {
H A Dweighted_picker.h112 static int LevelSize(int level) { return 1 << level; } function in class:tensorflow::random::WeightedPicker

Completed in 105 milliseconds