Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dquantile_ops.cc142 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesMakeQuantileSummariesOp()
205 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
210 int64 num_features_; member in tensorflow::BoostedTreesMakeQuantileSummariesOp
280 OP_REQUIRES_OK(context, context->GetAttr(kNumStreamsName, &num_features_)); in BoostedTreesQuantileStreamResourceDeserializeOp()
311 const int64 kCostPerUnit = 500 * num_features_; in Compute()
314 Shard(worker_threads.num_threads, worker_threads.workers, num_features_, in Compute()
319 int64 num_features_; member in tensorflow::BoostedTreesQuantileStreamResourceDeserializeOp
387 OP_REQUIRES_OK(context, context->GetAttr(kNumFeaturesName, &num_features_)); in BoostedTreesQuantileStreamResourceGetBucketBoundariesOp()
402 CHECK_EQ(num_features_, num_streams); in Compute()
431 int64 num_features_; member in tensorflow::BoostedTreesQuantileStreamResourceGetBucketBoundariesOp
[all …]
Dstats_ops.cc30 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesCalculateBestGainsPerFeatureOp()
85 for (int feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in Compute()
201 int num_features_; member in tensorflow::BoostedTreesCalculateBestGainsPerFeatureOp
214 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesMakeStatsSummaryOp()
241 {num_features_, max_splits_, num_buckets_, 2}, in Compute()
247 for (int feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in Compute()
269 int num_features_; member in tensorflow::BoostedTreesMakeStatsSummaryOp
Dtraining_ops.cc48 OP_REQUIRES_OK(context, context->GetAttr("num_features", &num_features_)); in BoostedTreesUpdateEnsembleOp()
202 for (int64 feature_idx = 0; feature_idx < num_features_; ++feature_idx) { in FindBestSplitsPerNode()
237 int32 num_features_; member in tensorflow::BoostedTreesUpdateEnsembleOp
/external/tensorflow/tensorflow/contrib/libsvm/kernels/
Ddecode_libsvm_op.cc30 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_features", &num_features_)); in DecodeLibsvmOp()
31 OP_REQUIRES(ctx, (num_features_ >= 1), in DecodeLibsvmOp()
33 num_features_, "\"")); in DecodeLibsvmOp()
133 shape(input_tensor->shape().dims()) = num_features_; in Compute()
137 int64 num_features_; member in tensorflow::DecodeLibsvmOp
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dtest_utils.h56 num_features_(num_features), in TestableDataSet()
60 return data_[example * num_features_ + feature_id]; in GetExampleValue()
64 int num_features_;
/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()