Home
last modified time | relevance | path

Searched refs:cuts_ptr (Results 1 – 10 of 10) sorted by relevance

/dports/misc/py-xgboost/xgboost-1.5.1/src/common/
H A Dhist_util.cuh39 cuts_ptr->SetDevice(device); in GetColumnSizesScan()
40 cuts_ptr->Resize(num_columns + 1, 0); in GetColumnSizesScan()
57 cuts_ptr->DevicePointer()); in GetColumnSizesScan()
151 cuts_ptr.SetDevice(device); in ProcessSlidingWindow()
154 &cuts_ptr, in ProcessSlidingWindow()
162 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessSlidingWindow()
167 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessSlidingWindow()
168 auto const &h_cuts_ptr = cuts_ptr.HostVector(); in ProcessSlidingWindow()
199 &cuts_ptr, in ProcessWeightedSlidingWindow()
239 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessWeightedSlidingWindow()
[all …]
H A Dquantile.cu58 common::Span<SketchContainer::OffsetT const> cuts_ptr, in PruneImpl() argument
65 size_t column_id = dh::SegmentId(cuts_ptr, idx); in PruneImpl()
67 cuts_ptr[column_id], cuts_ptr[column_id + 1] - cuts_ptr[column_id]); in PruneImpl()
71 auto to = cuts_ptr[column_id + 1] - cuts_ptr[column_id]; in PruneImpl()
72 idx -= cuts_ptr[column_id]; in PruneImpl()
311 common::Span<OffsetT> cuts_ptr, in Push() argument
350 auto n_uniques = this->ScanInput(out, cuts_ptr); in Push()
353 CHECK_EQ(this->columns_ptr_.Size(), cuts_ptr.size()); in Push()
355 this->Merge(cuts_ptr, out); in Push()
360 this->columns_ptr_.Resize(cuts_ptr.size()); in Push()
[all …]
H A Dhist_util.cu200 HostDeviceVector<SketchContainer::OffsetT> cuts_ptr; in ProcessBatch() local
211 &cuts_ptr, &column_sizes_scan); in ProcessBatch()
212 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessBatch()
219 auto const& h_cuts_ptr = cuts_ptr.ConstHostVector(); in ProcessBatch()
228 CHECK_NE(cuts_ptr.Size(), 0); in ProcessBatch()
269 HostDeviceVector<SketchContainer::OffsetT> cuts_ptr; in ProcessWeightedBatch() local
280 &cuts_ptr, &column_sizes_scan); in ProcessWeightedBatch()
281 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessWeightedBatch()
287 auto const& h_cuts_ptr = cuts_ptr.ConstHostVector(); in ProcessWeightedBatch()
H A Dquantile.cuh144 common::Span<OffsetT> cuts_ptr, size_t total_cuts,
/dports/misc/xgboost/xgboost-1.5.1/src/common/
H A Dhist_util.cuh39 cuts_ptr->SetDevice(device); in GetColumnSizesScan()
40 cuts_ptr->Resize(num_columns + 1, 0); in GetColumnSizesScan()
57 cuts_ptr->DevicePointer()); in GetColumnSizesScan()
151 cuts_ptr.SetDevice(device); in ProcessSlidingWindow()
154 &cuts_ptr, in ProcessSlidingWindow()
162 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessSlidingWindow()
167 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessSlidingWindow()
168 auto const &h_cuts_ptr = cuts_ptr.HostVector(); in ProcessSlidingWindow()
199 &cuts_ptr, in ProcessWeightedSlidingWindow()
239 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessWeightedSlidingWindow()
[all …]
H A Dquantile.cu58 common::Span<SketchContainer::OffsetT const> cuts_ptr, in PruneImpl() argument
65 size_t column_id = dh::SegmentId(cuts_ptr, idx); in PruneImpl()
67 cuts_ptr[column_id], cuts_ptr[column_id + 1] - cuts_ptr[column_id]); in PruneImpl()
71 auto to = cuts_ptr[column_id + 1] - cuts_ptr[column_id]; in PruneImpl()
72 idx -= cuts_ptr[column_id]; in PruneImpl()
311 common::Span<OffsetT> cuts_ptr, in Push() argument
350 auto n_uniques = this->ScanInput(out, cuts_ptr); in Push()
353 CHECK_EQ(this->columns_ptr_.Size(), cuts_ptr.size()); in Push()
355 this->Merge(cuts_ptr, out); in Push()
360 this->columns_ptr_.Resize(cuts_ptr.size()); in Push()
[all …]
H A Dhist_util.cu200 HostDeviceVector<SketchContainer::OffsetT> cuts_ptr; in ProcessBatch() local
211 &cuts_ptr, &column_sizes_scan); in ProcessBatch()
212 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessBatch()
219 auto const& h_cuts_ptr = cuts_ptr.ConstHostVector(); in ProcessBatch()
228 CHECK_NE(cuts_ptr.Size(), 0); in ProcessBatch()
269 HostDeviceVector<SketchContainer::OffsetT> cuts_ptr; in ProcessWeightedBatch() local
280 &cuts_ptr, &column_sizes_scan); in ProcessWeightedBatch()
281 auto d_cuts_ptr = cuts_ptr.DeviceSpan(); in ProcessWeightedBatch()
287 auto const& h_cuts_ptr = cuts_ptr.ConstHostVector(); in ProcessWeightedBatch()
H A Dquantile.cuh144 common::Span<OffsetT> cuts_ptr, size_t total_cuts,
/dports/misc/py-xgboost/xgboost-1.5.1/tests/cpp/common/
H A Dtest_quantile.cu14 dh::device_vector<bst_row_t> cuts_ptr(kCols+1); in TEST() local
15 thrust::fill(cuts_ptr.begin(), cuts_ptr.end(), 0); in TEST()
17 sketch.Push(dh::ToSpan(entries), dh::ToSpan(cuts_ptr), dh::ToSpan(cuts_ptr), 0); in TEST()
555 dh::device_vector<size_t> cuts_ptr(columns_ptr); in TEST() local
558 dh::ToSpan(cuts_ptr), kRows * kCols, {}); in TEST()
/dports/misc/xgboost/xgboost-1.5.1/tests/cpp/common/
H A Dtest_quantile.cu14 dh::device_vector<bst_row_t> cuts_ptr(kCols+1); in TEST() local
15 thrust::fill(cuts_ptr.begin(), cuts_ptr.end(), 0); in TEST()
17 sketch.Push(dh::ToSpan(entries), dh::ToSpan(cuts_ptr), dh::ToSpan(cuts_ptr), 0); in TEST()
555 dh::device_vector<size_t> cuts_ptr(columns_ptr); in TEST() local
558 dh::ToSpan(cuts_ptr), kRows * kCols, {}); in TEST()