| /external/libtextclassifier/ |
| D | quantization_test.cc | 47 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 …]
|
| D | quantization.cc | 23 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()
|
| D | model-executor.cc | 109 const int num_sparse_features = sparse_features.size(); in AddEmbedding() local
110 for (int i = 0; i < num_sparse_features; ++i) { in AddEmbedding()
117 bytes_per_embedding_, num_sparse_features, in AddEmbedding()
|
| D | quantization.h | 33 int bytes_per_embedding, int num_sparse_features,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_ops.cc | 85 context, context->GetAttr("num_sparse_features", &num_sparse_features)); in ComputeOptions()
91 context, num_sparse_features + num_dense_features > 0, in ComputeOptions()
95 static_cast<int64>(num_sparse_features) + in ComputeOptions()
100 static_cast<int64>(num_sparse_features) + in ComputeOptions()
111 int num_sparse_features = 0; member
130 examples.Initialize(context, model_weights, options.num_sparse_features, in DoCompute()
|
| D | sdca_internal.cc | 308 const int num_sparse_features, in Initialize() argument
311 num_features_ = num_sparse_features + num_dense_features; in Initialize()
352 example->sparse_features_.resize(num_sparse_features); in Initialize()
360 worker_threads, num_examples, num_sparse_features, weights, in Initialize()
367 num_sparse_features, num_dense_features, in Initialize()
374 const int num_sparse_features, const ModelWeights& weights, in CreateSparseFeatureRepresentation() argument
447 Shard(worker_threads.num_threads, worker_threads.workers, num_sparse_features, in CreateSparseFeatureRepresentation()
486 const int num_sparse_features, const int num_dense_features, in ComputeSquaredNormPerExample() argument
496 for (int j = 0; j < num_sparse_features; ++j) { in ComputeSquaredNormPerExample()
516 const int64 kCostPerUnit = num_dense_features + num_sparse_features; in ComputeSquaredNormPerExample()
|
| D | sdca_internal.h | 346 int num_sparse_features,
356 int num_sparse_features, const ModelWeights& weights,
375 int num_sparse_features, int num_dense_features,
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/ops/ |
| D | quantile_ops.cc | 189 int num_sparse_features; in __anon68c664fc0402() local
191 c->GetAttr("num_sparse_features", &num_sparse_features)); in __anon68c664fc0402()
193 int example_weights_index = num_dense_features + num_sparse_features * 3; in __anon68c664fc0402()
205 for (int i = 0; i < num_sparse_features; ++i) { in __anon68c664fc0402()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 147 int* num_sparse_features) { in ReadAndValidateAttributes() argument
151 context->GetAttr(kNumSparseFeaturesName, num_sparse_features)); in ReadAndValidateAttributes()
152 if ((*num_dense_features) + (*num_sparse_features) == 0) { in ReadAndValidateAttributes()
863 int num_sparse_features; in QuantilesOp() local
866 &num_sparse_features)); in QuantilesOp()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SdcaOptimizer.pbtxt | 105 name: "num_sparse_features"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 23146 number_attr: "num_sparse_features"
23151 number_attr: "num_sparse_features"
23174 number_attr: "num_sparse_features"
23179 number_attr: "num_sparse_features"
23197 number_attr: "num_sparse_features"
23224 name: "num_sparse_features"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 29537 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"
|
| D | ops_history.v1.pbtxt | 49293 number_attr: "num_sparse_features"
49298 number_attr: "num_sparse_features"
49321 number_attr: "num_sparse_features"
49326 number_attr: "num_sparse_features"
49344 number_attr: "num_sparse_features"
49371 name: "num_sparse_features"
|