| /external/libtextclassifier/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/core/kernels/ |
| D | string_to_hash_bucket_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()
92 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
96 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
| D | string_to_hash_bucket_op.cc | 47 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
51 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | grouping.py | 193 bucket_id = math_ops.reduce_min(array_ops.where(conditions_c))
195 return bucket_id
197 def window_size_fn(bucket_id): argument
199 window_size = batch_sizes[bucket_id]
213 def batching_fn(bucket_id, grouped_dataset): argument
215 batch_size = window_size_fn(bucket_id)
223 bucket_id,
230 bucket_boundary = boundaries[bucket_id]
|
| /external/tensorflow/tensorflow/examples/get_started/regression/ |
| D | imports85.py | 123 bucket_id = tf.string_to_hash_bucket_fast(line, num_buckets)
125 return bucket_id < int(train_fraction * num_buckets)
|