Searched defs:bucket_id (Results 1 – 13 of 13) 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
60 const uint64 bucket_id = input_hash % num_buckets; in operator() local
50 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
46 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
57 const uint64 bucket_id = input_hash % num_buckets_; in Compute() local
3065 def window_size_fn(bucket_id): argument3081 def batching_fn(bucket_id, grouped_dataset): argument
985 unsigned int bucket_id: 3; member41481 u32 bucket_id; member114114 u32 bucket_id; member115018 unsigned int bucket_id; member
601 unsigned int bucket_id: 3; member31040 u32 bucket_id; member100941 u32 bucket_id; member101839 unsigned int bucket_id; member
36097 u32 bucket_id; member81923 u32 bucket_id; member81961 unsigned int bucket_id; member