Lines Matching refs:num_columns

36   int num_columns = 1;  in TEST()  local
40 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
67 int num_columns = 100; in TEST() local
70 auto x = GenerateRandom(num_rows, num_columns); in TEST()
71 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
78 num_rows, num_columns, num_rows * num_columns, num_bins, false); in TEST()
85 int num_columns = 100; in TEST() local
88 auto x = GenerateRandom(num_rows, num_columns); in TEST()
89 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
98 num_rows, num_columns, num_rows * num_columns, num_bins, true); in TEST()
105 int num_columns = 5; in TEST() local
107 auto x = GenerateRandom(num_rows, num_columns); in TEST()
108 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
172 int num_columns = 5; in TEST() local
174 auto x = GenerateRandom(num_rows, num_columns); in TEST()
175 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
186 int num_columns = 5; in TEST() local
188 auto x = GenerateRandom(num_rows, num_columns); in TEST()
189 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
201 int num_columns = 5; in TEST() local
203 auto x = GenerateRandom(num_rows, num_columns); in TEST()
204 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
205 auto weighted_dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
225 int num_columns = 5; in TEST() local
227 auto x = GenerateRandom(num_rows, num_columns); in TEST()
228 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
235 auto x = GenerateRandom(num_rows * batches, num_columns); in TEST()
236 auto dmat = GetDMatrixFromData(x, num_rows * batches, num_columns); in TEST()
251 int num_columns =5; in TEST() local
253 auto x = GenerateRandom(num_rows, num_columns); in TEST()
256 GetExternalMemoryDMatrixFromData(x, num_rows, num_columns, 100, temp); in TEST()
268 int num_columns = 5; in TEST() local
271 auto x = GenerateRandom(num_rows, num_columns); in TEST()
272 auto dmat = GetExternalMemoryDMatrixFromData(x, num_rows, num_columns, 100, temp); in TEST()
294 void ValidateBatchedCuts(Adapter adapter, int num_bins, int num_columns, int num_rows, in ValidateBatchedCuts() argument
323 int num_columns = 100; in TEST() local
326 auto x = GenerateRandom(num_rows, num_columns); in TEST()
328 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
335 num_rows, num_columns, num_rows * num_columns, num_bins, false); in TEST()
341 int num_columns = 100; in TEST() local
344 auto x = GenerateRandom(num_rows, num_columns); in TEST()
346 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
353 SketchContainer sketch_container(ft, num_bins, num_columns, num_rows, 0); in TEST()
359 num_rows, num_columns, num_rows * num_columns, num_bins, false); in TEST()
366 int num_columns = 100; in TEST() local
369 auto x = GenerateRandom(num_rows, num_columns); in TEST()
371 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
381 SketchContainer sketch_container(ft, num_bins, num_columns, num_rows, 0); in TEST()
390 num_rows, num_columns, num_rows * num_columns, num_bins, true); in TEST()
462 int num_columns = 5; in TEST() local
464 auto x = GenerateRandom(num_rows, num_columns); in TEST()
465 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
468 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
469 ValidateBatchedCuts(adapter, num_bins, num_columns, num_rows, dmat.get()); in TEST()
478 int num_columns = 5; in TEST() local
480 auto x = GenerateRandom(num_rows, num_columns); in TEST()
481 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
483 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
484 ValidateBatchedCuts(adapter, num_bins, num_columns, num_rows, dmat.get(), batch_size); in TEST()
493 int num_columns = 5; in TEST() local
495 auto x = GenerateRandom(num_rows, num_columns); in TEST()
496 auto dmat = GetDMatrixFromData(x, num_rows, num_columns); in TEST()
500 auto adapter = AdapterFromData(x_device, num_rows, num_columns); in TEST()
507 ValidateBatchedCuts(adapter, num_bins, num_columns, num_rows, dmat.get()); in TEST()