Lines Matching refs:n_bins
23 …RunWithSeedsAndBins(kRows, [kRows, kCols, sparsity](int32_t seed, size_t n_bins, MetaInfo const& i… in TestSketchUnique() argument
25 SketchContainer sketch(ft, n_bins, kCols, kRows, 0); in TestSketchUnique()
33 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TestSketchUnique()
35 auto n_cuts = detail::RequiredSampleCutsPerColumn(n_bins, kRows); in TestSketchUnique()
102 RunWithSeedsAndBins(kRows, [=](int32_t seed, size_t n_bins, MetaInfo const& info) { in TEST() argument
104 SketchContainer sketch(ft, n_bins, kCols, kRows, 0); in TEST()
112 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TEST()
114 auto n_cuts = detail::RequiredSampleCutsPerColumn(n_bins, kRows); in TEST()
119 sketch.Prune(n_bins); in TEST()
132 size_t n_bins = 10; in TEST() local
134 SketchContainer sketch_0(ft, n_bins, kCols, kRows, 0); in TEST()
141 AdapterDeviceSketch(adapter_0.Value(), n_bins, info, in TEST()
172 RunWithSeedsAndBins(kRows, [=](int32_t seed, size_t n_bins, MetaInfo const& info) { in TEST() argument
174 SketchContainer sketch_0(ft, n_bins, kCols, kRows, 0); in TEST()
181 AdapterDeviceSketch(adapter_0.Value(), n_bins, info, in TEST()
184 SketchContainer sketch_1(ft, n_bins, kCols, kRows * kRows, 0); in TEST()
191 AdapterDeviceSketch(adapter_1.Value(), n_bins, info, in TEST()
217 void TestMergeDuplicated(int32_t n_bins, size_t cols, size_t rows, float frac) { in TestMergeDuplicated() argument
221 SketchContainer sketch_0(ft, n_bins, cols, rows, 0); in TestMergeDuplicated()
228 AdapterDeviceSketch(adapter_0.Value(), n_bins, info, in TestMergeDuplicated()
233 SketchContainer sketch_1(ft, n_bins, cols, f_rows, 0); in TestMergeDuplicated()
254 AdapterDeviceSketch(adapter_1.Value(), n_bins, info, in TestMergeDuplicated()
274 size_t n_bins = 256; in TEST() local
277 TestMergeDuplicated(n_bins, kRows, kCols, frac); in TEST()
284 RunWithSeedsAndBins(kRows, [=](int32_t seed, size_t n_bins, in TEST()
288 SketchContainer sketch_on_single_node(ft, n_bins, kCols, kRows, 0); in TEST()
291 kRows * world, static_cast<size_t>(n_bins * WQSketch::kFactor)); in TEST()
301 containers.emplace_back(ft, n_bins, kCols, kRows, 0); in TEST()
302 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TEST()
334 RunWithSeedsAndBins(kRows, [=](int32_t seed, size_t n_bins, MetaInfo const& info) { in TEST() argument
337 SketchContainer sketch_on_single_node(ft, n_bins, kCols, kRows, 0); in TEST()
340 kRows * world, static_cast<size_t>(n_bins * WQSketch::kFactor)); in TEST()
350 containers.emplace_back(ft, n_bins, kCols, kRows, 0); in TEST()
351 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TEST()
367 SketchContainer sketch_distributed(ft, n_bins, kCols, kRows, 0); in TEST()
374 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TEST()
423 RunWithSeedsAndBins(kRows, [=](int32_t seed, size_t n_bins, in TEST()
427 SketchContainer sketch_distributed(ft, n_bins, kCols, kRows, 0); in TEST()
434 AdapterDeviceSketch(adapter.Value(), n_bins, info, in TEST()
491 int32_t n_bins = 128; in TEST() local
506 SketchContainer sketch(ft, n_bins, kCols, kRows, 0); in TEST()
544 int32_t n_bins = 16; in TEST() local
546 SketchContainer sketch(ft, n_bins, kCols, kRows, 0); in TEST()