Searched defs:bucket_id (Results 1 – 6 of 6) sorted by relevance
49 const int bucket_id = 0; in TEST() local67 const int bucket_id = 1; in TEST() local90 const int bucket_id = 1; in TEST() local113 const int bucket_id = 1; in TEST() local133 const int bucket_id = 1; in TEST() local
30 const int bucket_id, float* dest, int dest_size) { in DequantizeAdd8bit()42 int quantization_bits, int bucket_id, float* dest, in DequantizeAddNBit()75 int quantization_bits, int bucket_id, float* dest, in DequantizeAdd()
161 const int bucket_id = sparse_features.data()[i]; in AddEmbedding() local
50 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local92 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
47 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
197 def window_size_fn(bucket_id): argument213 def batching_fn(bucket_id, grouped_dataset): argument