Lines Matching refs:bucket_id
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()
149 expected.push_back(1.0 / num_sparse_features * (1 - 4) * scales[bucket_id]); in TEST()
150 expected.push_back(1.0 / num_sparse_features * (2 - 4) * scales[bucket_id]); in TEST()
151 expected.push_back(1.0 / num_sparse_features * (3 - 4) * scales[bucket_id]); in TEST()
152 expected.push_back(1.0 / num_sparse_features * (4 - 4) * scales[bucket_id]); in TEST()
153 expected.push_back(1.0 / num_sparse_features * (5 - 4) * scales[bucket_id]); in TEST()
154 expected.push_back(1.0 / num_sparse_features * (6 - 4) * scales[bucket_id]); in TEST()
155 expected.push_back(1.0 / num_sparse_features * (7 - 4) * scales[bucket_id]); in TEST()
156 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()
157 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()
158 expected.push_back(1.0 / num_sparse_features * (0 - 4) * scales[bucket_id]); in TEST()