| /external/libtextclassifier/native/annotator/ |
| D | quantization_test.cc | 49 const int bucket_id = 0; in TEST() local
52 num_sparse_features, quantization_bits, bucket_id, in TEST()
67 const int bucket_id = 1; in TEST() local
70 num_sparse_features, quantization_bits, bucket_id, in TEST()
90 const int bucket_id = 1; in TEST() local
99 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
113 const int bucket_id = 1; in TEST() local
120 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
133 const int bucket_id = 1; in TEST() local
145 num_sparse_features, quantization_bits, bucket_id, dest.data(), in TEST()
[all …]
|
| D | quantization.cc | 30 const int bucket_id, float* dest, int dest_size) { in DequantizeAdd8bit() argument
32 const float multiplier = scales[bucket_id]; in DequantizeAdd8bit()
36 embeddings[bucket_id * bytes_per_embedding + k]); in DequantizeAdd8bit()
42 int quantization_bits, int bucket_id, float* dest, in DequantizeAddNBit() argument
45 const float multiplier = scales[bucket_id]; in DequantizeAddNBit()
50 uint16 data = embeddings[bucket_id * bytes_per_embedding + read16_offset]; in DequantizeAddNBit()
54 data |= embeddings[bucket_id * bytes_per_embedding + read16_offset + 1] in DequantizeAddNBit()
75 int quantization_bits, int bucket_id, float* dest, 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 | 129 int TFLiteEmbeddingExecutor::PruneBucketId(int bucket_id) const { in PruneBucketId()
137 const int bucket_id_major = bucket_id >> 6; in PruneBucketId()
138 const int bucket_id_minor = bucket_id & 63; in PruneBucketId()
161 const int bucket_id = sparse_features.data()[i]; in AddEmbedding() local
168 if (bucket_id >= full_num_buckets) { in AddEmbedding()
173 final_bucket_id = PruneBucketId(bucket_id); in AddEmbedding()
175 final_bucket_id = bucket_id; in AddEmbedding()
|
| D | quantization.h | 34 int quantization_bits, int bucket_id, float* dest,
|
| D | model-executor.h | 94 int PruneBucketId(int bucket_id) const;
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 219 bucket_id = math_ops.reduce_min(array_ops.where(conditions_c))
221 return bucket_id
223 def window_size_fn(bucket_id): argument
225 window_size = batch_sizes[bucket_id]
239 def batching_fn(bucket_id, grouped_dataset): argument
241 batch_size = window_size_fn(bucket_id)
249 bucket_id,
256 bucket_boundary = boundaries[bucket_id]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | string_to_hash_bucket_fast_op.h | 50 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
54 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
| D | string_to_hash_bucket_op.cc | 46 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
50 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
| D | string_to_hash_bucket_op.h | 57 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
61 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1026 const int32 bucket_id = stats_summary_indices(idx, 2); in Compute() local
1033 bucket_id, std::vector<float>(stats_dims))); in Compute()
1171 const int32 bucket_id = b_iter->first; in process_node() local
1173 if (bucket_id == num_buckets) { in process_node()
1194 best_bucket = bucket_id; in process_node()
1212 best_bucket = bucket_id; in process_node()
1394 const int32 bucket_id) in StatsPartitionKey()
1395 : node_id(node_id), feature_dim(feature_dim), bucket_id(bucket_id) {} in StatsPartitionKey()
1399 (bucket_id == other.bucket_id); in operator ==()
1413 (a.bucket_id < b.bucket_id)) { in operator ()()
[all …]
|