HomeSort by relevance Sort by last modified time
    Searched refs:num_dense_features (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/boosted_trees/ops/
quantile_ops.cc 176 .Attr("num_dense_features: int >= 0")
179 .Input("dense_float_features: num_dense_features * float")
184 .Output("dense_summaries: num_dense_features * string")
187 int num_dense_features;
188 TF_RETURN_IF_ERROR(c->GetAttr("num_dense_features", &num_dense_features));
193 int example_weights_index = num_dense_features + num_sparse_features * 3;
196 for (int i = 0; i < num_dense_features; ++i) {
206 c->set_output(i + num_dense_features, c->Scalar());
213 num_dense_features: Number of dense feature groups to compute quantiles on
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
sdca_ops.cc 89 context->GetAttr("num_dense_features", &num_dense_features));
91 context, num_sparse_features + num_dense_features > 0,
96 static_cast<int64>(num_dense_features) <=
101 static_cast<int64>(num_dense_features),
113 int num_dense_features = 0; member in struct:tensorflow::__anon39911::ComputeOptions
132 options.num_dense_features));
sdca_internal.cc 310 const int num_dense_features) {
311 num_features_ = num_sparse_features + num_dense_features;
353 example->dense_vectors_.resize(num_dense_features);
364 worker_threads, num_examples, num_dense_features, weights,
367 num_sparse_features, num_dense_features,
454 const int num_dense_features, const ModelWeights& weights,
479 Shard(worker_threads.num_threads, worker_threads.workers, num_dense_features,
486 const int num_sparse_features, const int num_dense_features,
507 for (int j = 0; j < num_dense_features; ++j) {
516 const int64 kCostPerUnit = num_dense_features + num_sparse_features
    [all...]
sdca_internal.h 348 int num_dense_features);
367 int num_dense_features, const ModelWeights& weights,
375 int num_sparse_features, int num_dense_features,
  /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
quantile_ops.cc 63 const char* const kNumDenseFeaturesName = "num_dense_features";
146 int* num_dense_features,
149 context->GetAttr(kNumDenseFeaturesName, num_dense_features));
152 if ((*num_dense_features) + (*num_sparse_features) == 0) {
862 int num_dense_features; local
    [all...]

Completed in 281 milliseconds