Home
last modified time | relevance | path

Searched refs:num_features_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc143 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesMakeQuantileSummariesOp()
206 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
211 int64 num_features_; member in tensorflow::BoostedTreesMakeQuantileSummariesOp
223 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesFlushQuantileSummariesOp()
262 const int64 kCostPerUnit = 500 * num_features_; in Compute()
265 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
271 int64 num_features_; member in tensorflow::BoostedTreesFlushQuantileSummariesOp
346 OP_REQUIRES_OK(context, context->GetAttr(kNumStreamsName, &num_features_)); in BoostedTreesQuantileStreamResourceDeserializeOp()
376 const int64 kCostPerUnit = 500 * num_features_; in Compute()
379 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
[all …]
Dstats_ops.cc47 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
108 for (int feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in Compute()
225 int num_features_; member in tensorflow::BoostedTreesCalculateBestGainsPerFeatureOp
563 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesCalculateBestFeatureSplitV2()
592 DCHECK_EQ(stats_summaries_list.size(), num_features_); in Compute()
593 stats_summaries.reserve(num_features_); in Compute()
602 DCHECK_EQ(split_types.size(), num_features_); in Compute()
604 for (int i = 0; i < num_features_; ++i) { in Compute()
618 DCHECK_EQ(candidate_feature_ids.size(), num_features_); in Compute()
677 for (int f_idx = 0; f_idx < num_features_; ++f_idx) { in Compute()
[all …]
Dtraining_ops.cc38 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesUpdateEnsembleOp()
178 for (int64 feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in FindBestSplitsPerNode()
225 int32 num_features_; member in tensorflow::BoostedTreesUpdateEnsembleOp
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.h339 int num_features() const { return num_features_; } in num_features()
383 int num_features_ = 0; variable
Dsdca_internal.cc325 num_features_ = num_sparse_features + num_dense_features; in Initialize()