Home
last modified time | relevance | path

Searched refs:smallest_weight_class_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dgrow_stats.cc638 if (c == smallest_weight_class_) { in accumulate()
639 smallest_weight_class_ = argmin(class_weights_); in accumulate()
649 smallest_weight_class_ = argmin(class_weights_); in accumulate()
658 it = class_weights_.find(smallest_weight_class_); in accumulate()
662 smallest_weight_class_ = argmin(class_weights_); in accumulate()
671 auto it = class_weights_.find(smallest_weight_class_); in get_weight()
688 auto it = class_weights_.find(smallest_weight_class_); in set_sum_and_square()
710 smallest_weight_class_ = argmin(class_weights_); in ExtractFromProto()
Dgrow_stats.h427 : n_(n), num_classes_(num_classes), smallest_weight_class_(-1) {} in FixedSizeClassStats()
453 int smallest_weight_class_; variable