Method num_elements

tensorflow/core/lib/random/weighted_picker.h:128–128

inline int32 WeightedPicker::total_weight() const { return level_[0][0]; }

inline int WeightedPicker::num_elements() const { return N_; }

}  // namespace random
}  // namespace tensorflow