Home
last modified time | relevance | path

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

/external/libtextclassifier/native/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/core/ops/compat/ops_history_v2/
DSdcaOptimizer.pbtxt6 number_attr: "num_sparse_features"
11 number_attr: "num_sparse_features"
34 number_attr: "num_sparse_features"
39 number_attr: "num_sparse_features"
57 number_attr: "num_sparse_features"
84 name: "num_sparse_features"
124 number_attr: "num_sparse_features"
129 number_attr: "num_sparse_features"
152 number_attr: "num_sparse_features"
157 number_attr: "num_sparse_features"
[all …]
DSdcaOptimizerV2.pbtxt6 number_attr: "num_sparse_features"
11 number_attr: "num_sparse_features"
34 number_attr: "num_sparse_features"
39 number_attr: "num_sparse_features"
57 number_attr: "num_sparse_features"
85 name: "num_sparse_features"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DSdcaOptimizer.pbtxt6 number_attr: "num_sparse_features"
11 number_attr: "num_sparse_features"
34 number_attr: "num_sparse_features"
39 number_attr: "num_sparse_features"
57 number_attr: "num_sparse_features"
84 name: "num_sparse_features"
124 number_attr: "num_sparse_features"
129 number_attr: "num_sparse_features"
152 number_attr: "num_sparse_features"
157 number_attr: "num_sparse_features"
[all …]
DSdcaOptimizerV2.pbtxt6 number_attr: "num_sparse_features"
11 number_attr: "num_sparse_features"
34 number_attr: "num_sparse_features"
39 number_attr: "num_sparse_features"
57 number_attr: "num_sparse_features"
85 name: "num_sparse_features"
/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops.cc93 context, context->GetAttr("num_sparse_features", &num_sparse_features)); in ComputeOptions()
99 context, num_sparse_features + num_dense_features > 0, in ComputeOptions()
103 static_cast<int64>(num_sparse_features) + in ComputeOptions()
108 static_cast<int64>(num_sparse_features) + in ComputeOptions()
119 int num_sparse_features = 0; member
138 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
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_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/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.pbtxt44836 number_attr: "num_sparse_features"
44841 number_attr: "num_sparse_features"
44864 number_attr: "num_sparse_features"
44869 number_attr: "num_sparse_features"
44887 number_attr: "num_sparse_features"
44915 name: "num_sparse_features"
44955 number_attr: "num_sparse_features"
44960 number_attr: "num_sparse_features"
44983 number_attr: "num_sparse_features"
44988 number_attr: "num_sparse_features"
[all …]
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt66501 number_attr: "num_sparse_features"
66506 number_attr: "num_sparse_features"
66529 number_attr: "num_sparse_features"
66534 number_attr: "num_sparse_features"
66552 number_attr: "num_sparse_features"
66579 name: "num_sparse_features"
66619 number_attr: "num_sparse_features"
66624 number_attr: "num_sparse_features"
66647 number_attr: "num_sparse_features"
66652 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"