Home
last modified time | relevance | path

Searched refs:num_sparse_features (Results 1 – 15 of 15) sorted by relevance

/external/libtextclassifier/annotator/
Dquantization_test.cc47 const int num_sparse_features = 7; in TEST() local
52 num_sparse_features, quantization_bits, bucket_id, in TEST()
70 num_sparse_features, quantization_bits, bucket_id, in TEST()
88 const int num_sparse_features = 7; in TEST() local
99 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
104 1.0 / num_sparse_features * (0 - 1)); in TEST()
111 const int num_sparse_features = 7; in TEST() local
120 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
124 1.0 / num_sparse_features * (1 - 1)); in TEST()
131 const int num_sparse_features = 7; in TEST() local
[all …]
Dquantization.cc23 float DequantizeValue(int num_sparse_features, int quantization_bias, in DequantizeValue() argument
25 return 1.0 / num_sparse_features * (value - quantization_bias) * multiplier; in DequantizeValue()
29 int bytes_per_embedding, const int num_sparse_features, in DequantizeAdd8bit() argument
35 DequantizeValue(num_sparse_features, kQuantizationBias8bit, multiplier, in DequantizeAdd8bit()
41 int bytes_per_embedding, int num_sparse_features, in DequantizeAddNBit() argument
58 dest[i] += DequantizeValue(num_sparse_features, quantization_bias, in DequantizeAddNBit()
74 int bytes_per_embedding, int num_sparse_features, in DequantizeAdd() argument
79 num_sparse_features, bucket_id, dest, dest_size); in DequantizeAdd()
82 num_sparse_features, quantization_bits, bucket_id, dest, in DequantizeAdd()
Dmodel-executor.cc159 const int num_sparse_features = sparse_features.size(); in AddEmbedding() local
160 for (int i = 0; i < num_sparse_features; ++i) { in AddEmbedding()
178 bytes_per_embedding_, num_sparse_features, in AddEmbedding()
Dquantization.h33 int bytes_per_embedding, int num_sparse_features,
/external/tensorflow/tensorflow/contrib/boosted_trees/ops/
Dquantile_ops.cc189 int num_sparse_features; in __anonf41644880402() local
191 c->GetAttr("num_sparse_features", &num_sparse_features)); in __anonf41644880402()
193 int example_weights_index = num_dense_features + num_sparse_features * 3; in __anonf41644880402()
205 for (int i = 0; i < num_sparse_features; ++i) { in __anonf41644880402()
278 int num_sparse_features; in __anonf41644880502() local
280 c->GetAttr("num_sparse_features", &num_sparse_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.cc322 const int num_sparse_features, in Initialize() argument
325 num_features_ = num_sparse_features + num_dense_features; in Initialize()
366 example->sparse_features_.resize(num_sparse_features); in Initialize()
374 worker_threads, num_examples, num_sparse_features, weights, in Initialize()
381 worker_threads, num_examples, num_sparse_features, num_dense_features, in Initialize()
388 const int num_sparse_features, const ModelWeights& weights, in CreateSparseFeatureRepresentation() argument
461 Shard(worker_threads.num_threads, worker_threads.workers, num_sparse_features, in CreateSparseFeatureRepresentation()
500 const int num_sparse_features, const int num_dense_features, in ComputeSquaredNormPerExample() argument
513 for (int j = 0; j < num_sparse_features; ++j) { in ComputeSquaredNormPerExample()
540 const int64 kCostPerUnit = num_dense_features + num_sparse_features; in ComputeSquaredNormPerExample()
Dsdca_ops.cc92 context, context->GetAttr("num_sparse_features", &num_sparse_features)); in ComputeOptions()
98 context, num_sparse_features + num_dense_features > 0, in ComputeOptions()
102 static_cast<int64>(num_sparse_features) + in ComputeOptions()
107 static_cast<int64>(num_sparse_features) + in ComputeOptions()
118 int num_sparse_features = 0; member
137 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
Dsdca_internal.h343 int num_sparse_features,
353 int num_sparse_features, const ModelWeights& weights,
372 int num_sparse_features, int num_dense_features,
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dquantile_ops.cc149 int* num_sparse_features) { in ReadAndValidateAttributes() argument
153 context->GetAttr(kNumSparseFeaturesName, num_sparse_features)); in ReadAndValidateAttributes()
154 if ((*num_dense_features) + (*num_sparse_features) == 0) { in ReadAndValidateAttributes()
872 int num_sparse_features; in QuantilesOp() local
875 &num_sparse_features)); in QuantilesOp()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SdcaOptimizerV2.pbtxt109 name: "num_sparse_features"
Dapi_def_SdcaOptimizer.pbtxt108 name: "num_sparse_features"
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt33476 number_attr: "num_sparse_features"
33481 number_attr: "num_sparse_features"
33504 number_attr: "num_sparse_features"
33509 number_attr: "num_sparse_features"
33527 number_attr: "num_sparse_features"
33555 name: "num_sparse_features"
33595 number_attr: "num_sparse_features"
33600 number_attr: "num_sparse_features"
33623 number_attr: "num_sparse_features"
33628 number_attr: "num_sparse_features"
[all …]
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt67164 number_attr: "num_sparse_features"
67169 number_attr: "num_sparse_features"
67192 number_attr: "num_sparse_features"
67197 number_attr: "num_sparse_features"
67215 number_attr: "num_sparse_features"
67242 name: "num_sparse_features"
67282 number_attr: "num_sparse_features"
67287 number_attr: "num_sparse_features"
67310 number_attr: "num_sparse_features"
67315 number_attr: "num_sparse_features"
[all …]
Dops_history.v1.pbtxt67791 number_attr: "num_sparse_features"
67796 number_attr: "num_sparse_features"
67819 number_attr: "num_sparse_features"
67824 number_attr: "num_sparse_features"
67842 number_attr: "num_sparse_features"
67869 name: "num_sparse_features"
67909 number_attr: "num_sparse_features"
67914 number_attr: "num_sparse_features"
67937 number_attr: "num_sparse_features"
67942 number_attr: "num_sparse_features"
[all …]
Dops_history.v0.pbtxt29537 number_attr: "num_sparse_features"
29542 number_attr: "num_sparse_features"
29565 number_attr: "num_sparse_features"
29570 number_attr: "num_sparse_features"
29588 number_attr: "num_sparse_features"
29615 name: "num_sparse_features"