Home
last modified time | relevance | path

Searched refs:FixedSizeClassStats (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats_test.cc32 using tensorflow::tensorforest::FixedSizeClassStats;
353 TEST(FixedSizeClassStats, Exact) { in TEST() argument
354 FixedSizeClassStats stats(10, 100); in TEST()
372 TEST(FixedSizeClassStats, Approximate) { in TEST() argument
373 FixedSizeClassStats stats(5, 10); in TEST()
Dgrow_stats.h423 class FixedSizeClassStats {
426 FixedSizeClassStats(int n, int num_classes) in FixedSizeClassStats() function
475 FixedSizeClassStats stats(params_.num_classes_to_track(), in ClassificationAddSplitStats()
518 std::vector<FixedSizeClassStats> left_counts_;
519 std::vector<FixedSizeClassStats> right_counts_;
Dgrow_stats.cc634 void FixedSizeClassStats::accumulate(int c, float w) { in accumulate()
665 float FixedSizeClassStats::get_weight(int c) const { in get_weight()
683 void FixedSizeClassStats::set_sum_and_square(float* sum, float* square) const { in set_sum_and_square()
704 void FixedSizeClassStats::ExtractFromProto( in ExtractFromProto()
714 void FixedSizeClassStats::PackToProto( in PackToProto()