Home
last modified time | relevance | path

Searched refs:get_weight (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/lib/random/
Dweighted_picker.h61 int32 get_weight(int index) const;
120 inline int32 WeightedPicker::get_weight(int index) const { in get_weight() function
Dweighted_picker.cc138 const int32 delta = weight - get_weight(index); in set_weight()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats_test.cc360 EXPECT_EQ(stats.get_weight(1), 1.0); in TEST()
361 EXPECT_EQ(stats.get_weight(2), 2.0); in TEST()
362 EXPECT_EQ(stats.get_weight(3), 3.0); in TEST()
383 float diff = stats.get_weight(i) - i * 1.0; in TEST()
Dgrow_stats.h435 float get_weight(int c) const;
510 return left_counts_[split].get_weight(class_num); in left_count()
514 return right_counts_[split].get_weight(class_num); in right_count()
Dgrow_stats.cc665 float FixedSizeClassStats::get_weight(int c) const { in get_weight() function in tensorflow::tensorforest::FixedSizeClassStats
696 w = get_weight(it.first); in set_sum_and_square()
/external/tensorflow/tensorflow/core/kernels/
Drange_sampler.cc187 return static_cast<float>(picker_.get_weight(value)) / picker_.total_weight(); in Probability()
195 picker_.set_weight(value, picker_.get_weight(value) + 1); in Update()