Home
last modified time | relevance | path

Searched refs:feature_idx (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/utils/
Dexample.h34 void Add(const int32 feature_idx, const T value) { in Add() argument
35 values_.emplace_back(feature_idx, value); in Add()
42 OptionalValue<T> operator[](int feature_idx) const {
45 [&feature_idx](const std::pair<int32, T>& element) {
46 return element.first == feature_idx;
75 void Add(const int32 feature_idx, const float value) { in Add() argument
77 DCHECK_EQ(0, feature_idx); in Add()
80 multidimensional_values.Add(feature_idx, value); in Add()
91 OptionalValue<T> operator[](int feature_idx) const {
98 return multidimensional_values[feature_idx];
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtraining_ops.cc32 int64 feature_idx; member
118 const int64 feature_idx = candidate.feature_idx; in Compute() local
121 const int32 feature_id = feature_ids(feature_idx); in Compute()
123 thresholds_list[feature_idx].vec<int32>()(candidate_idx); in Compute()
124 const float gain = gains_list[feature_idx].vec<float>()(candidate_idx); in Compute()
137 left_node_contribs[feature_idx].matrix<float>()(candidate_idx, 0); in Compute()
140 right_node_contribs[feature_idx].matrix<float>()(candidate_idx, 0); in Compute()
202 for (int64 feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in FindBestSplitsPerNode() local
203 const auto& node_ids = node_ids_list[feature_idx].vec<int32>(); in FindBestSplitsPerNode()
204 const auto& gains = gains_list[feature_idx].vec<float>(); in FindBestSplitsPerNode()
[all …]
Dstats_ops.cc85 for (int feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in Compute() local
104 total_grad += stats_summary[feature_idx](node_id, bucket, 0); in Compute()
105 total_hess += stats_summary[feature_idx](node_id, bucket, 1); in Compute()
157 output_node_ids_list.allocate(feature_idx, {num_nodes}, in Compute()
163 feature_idx, {num_nodes}, &output_gains_t)); in Compute()
168 output_thresholds_list.allocate(feature_idx, {num_nodes}, in Compute()
174 feature_idx, {num_nodes, 1}, in Compute()
181 feature_idx, {num_nodes, 1}, in Compute()
247 for (int feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in Compute() local
248 const auto& features = bucketized_features_list[feature_idx].vec<int32>(); in Compute()
[all …]
Dquantile_ops.cc242 for (int64 feature_idx = begin; feature_idx < end; ++feature_idx) { in Compute() local
243 const Tensor& summaries = summaries_list[feature_idx]; in Compute()
258 stream_resource->stream(feature_idx)->PushSummary(summary_entries); in Compute()
467 for (int64 feature_idx = begin; feature_idx < end; feature_idx++) { in Compute() local
468 const Tensor& values_tensor = float_features_list[feature_idx]; in Compute()
474 feature_idx, TensorShape({num_values}), &output_t)); in Compute()
478 GetBuckets(feature_idx, bucket_boundaries_list); in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dsplit_handler_ops.cc840 for (int64 feature_idx = start_index + 1; feature_idx < end_index; in ComputeNormalDecisionTree() local
841 ++feature_idx) { in ComputeNormalDecisionTree()
843 feature_idx); in ComputeNormalDecisionTree()
854 best_feature_idx = feature_idx; in ComputeNormalDecisionTree()
/external/libvpx/libvpx/vp9/encoder/
Dvp9_encodeframe.c3539 int feature_idx = 0; in ml_predict_var_rd_paritioning() local
3543 features[feature_idx++] = logf((float)(dc_q * dc_q) / 256.0f + 1.0f); in ml_predict_var_rd_paritioning()
3559 features[feature_idx++] = logf((float)var + 1.0f); in ml_predict_var_rd_paritioning()
3570 features[feature_idx++] = var_ratio; in ml_predict_var_rd_paritioning()
3573 assert(feature_idx == FEATURES); in ml_predict_var_rd_paritioning()
4576 int feature_idx = 0; in ml_predict_var_paritioning() local
4579 features[feature_idx++] = logf((float)(dc_q * dc_q) / 256.0f + 1.0f); in ml_predict_var_paritioning()
4597 features[feature_idx++] = logf((float)var + 1.0f); in ml_predict_var_paritioning()
4608 features[feature_idx++] = var_ratio; in ml_predict_var_paritioning()
4612 assert(feature_idx == FEATURES); in ml_predict_var_paritioning()
/external/libaom/libaom/av1/encoder/
Drdopt.c5089 int feature_idx = 0; in get_mean_dev_features() local
5103 feature[feature_idx++] = mean; in get_mean_dev_features()
5104 feature[feature_idx++] = dev; in get_mean_dev_features()
5113 feature[feature_idx++] = dev; in get_mean_dev_features()
5116 feature[feature_idx++] = mean; in get_mean_dev_features()