Home
last modified time | relevance | path

Searched refs:num_dense_features (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dquantile_ops.cc187 int num_dense_features; in __anonf41644880402() local
188 TF_RETURN_IF_ERROR(c->GetAttr("num_dense_features", &num_dense_features)); in __anonf41644880402()
193 int example_weights_index = num_dense_features + num_sparse_features * 3; in __anonf41644880402()
196 for (int i = 0; i < num_dense_features; ++i) { in __anonf41644880402()
206 c->set_output(i + num_dense_features, c->Scalar()); in __anonf41644880402()
276 int num_dense_features; in __anonf41644880502() local
277 TF_RETURN_IF_ERROR(c->GetAttr("num_dense_features", &num_dense_features)); in __anonf41644880502()
284 for (int i = 0; i < num_dense_features + num_sparse_features; ++i) { in __anonf41644880502()
/external/tensorflow/tensorflow/core/kernels/
Dsdca_internal.cc324 const int num_dense_features) { in Initialize() argument
325 num_features_ = num_sparse_features + num_dense_features; in Initialize()
367 example->dense_vectors_.resize(num_dense_features); in Initialize()
378 worker_threads, num_examples, num_dense_features, weights, in Initialize()
381 worker_threads, num_examples, num_sparse_features, num_dense_features, in Initialize()
468 const int num_dense_features, const ModelWeights& weights, in CreateDenseFeatureRepresentation() argument
493 Shard(worker_threads.num_threads, worker_threads.workers, num_dense_features, in CreateDenseFeatureRepresentation()
500 const int num_sparse_features, const int num_dense_features, in ComputeSquaredNormPerExample() argument
531 for (int j = 0; j < num_dense_features; ++j) { in ComputeSquaredNormPerExample()
540 const int64 kCostPerUnit = num_dense_features + num_sparse_features; in ComputeSquaredNormPerExample()
Dsdca_ops.cc96 context->GetAttr("num_dense_features", &num_dense_features)); in ComputeOptions()
98 context, num_sparse_features + num_dense_features > 0, in ComputeOptions()
103 static_cast<int64>(num_dense_features) <= in ComputeOptions()
108 static_cast<int64>(num_dense_features), in ComputeOptions()
120 int num_dense_features = 0; member
139 options.num_dense_features)); in DoCompute()
Dsdca_internal.h345 int num_dense_features);
364 int num_dense_features, const ModelWeights& weights,
372 int num_sparse_features, int num_dense_features,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc148 int* num_dense_features, in ReadAndValidateAttributes() argument
151 context->GetAttr(kNumDenseFeaturesName, num_dense_features)); in ReadAndValidateAttributes()
154 if ((*num_dense_features) + (*num_sparse_features) == 0) { in ReadAndValidateAttributes()
871 int num_dense_features; in QuantilesOp() local
874 ReadAndValidateAttributes(context, &num_dense_features, in QuantilesOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SdcaOptimizerV2.pbtxt122 name: "num_dense_features"
Dapi_def_SdcaOptimizer.pbtxt121 name: "num_dense_features"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt33491 number_attr: "num_dense_features"
33514 number_attr: "num_dense_features"
33532 number_attr: "num_dense_features"
33565 name: "num_dense_features"
33610 number_attr: "num_dense_features"
33633 number_attr: "num_dense_features"
33651 number_attr: "num_dense_features"
33684 name: "num_dense_features"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt67179 number_attr: "num_dense_features"
67202 number_attr: "num_dense_features"
67220 number_attr: "num_dense_features"
67252 name: "num_dense_features"
67297 number_attr: "num_dense_features"
67320 number_attr: "num_dense_features"
67338 number_attr: "num_dense_features"
67371 name: "num_dense_features"
67416 number_attr: "num_dense_features"
67439 number_attr: "num_dense_features"
[all …]
Dops_history.v1.pbtxt67806 number_attr: "num_dense_features"
67829 number_attr: "num_dense_features"
67847 number_attr: "num_dense_features"
67879 name: "num_dense_features"
67924 number_attr: "num_dense_features"
67947 number_attr: "num_dense_features"
67965 number_attr: "num_dense_features"
67998 name: "num_dense_features"
68043 number_attr: "num_dense_features"
68066 number_attr: "num_dense_features"
[all …]
Dops_history.v0.pbtxt29552 number_attr: "num_dense_features"
29575 number_attr: "num_dense_features"
29593 number_attr: "num_dense_features"
29625 name: "num_dense_features"