|
| SketchContainerImpl (Context const *ctx, std::vector< bst_row_t > columns_size, int32_t max_bins, common::Span< FeatureType const > feature_types, bool use_group) |
|
void | GatherSketchInfo (MetaInfo const &info, std::vector< typename WQSketch::SummaryContainer > const &reduced, std::vector< bst_row_t > *p_worker_segments, std::vector< bst_row_t > *p_sketches_scan, std::vector< typename WQSketch::Entry > *p_global_sketches) |
|
void | AllReduce (MetaInfo const &info, std::vector< typename WQSketch::SummaryContainer > *p_reduced, std::vector< int32_t > *p_num_cuts) |
|
template<typename Batch , typename IsValid > |
void | PushRowPageImpl (Batch const &batch, size_t base_rowid, OptionalWeights weights, size_t nnz, size_t n_features, bool is_dense, IsValid is_valid) |
|
void | PushRowPage (SparsePage const &page, MetaInfo const &info, Span< float const > hessian={}) |
|
void | MakeCuts (MetaInfo const &info, HistogramCuts *cuts) |
|
template<typename WQSketch>
class xgboost::common::SketchContainerImpl< WQSketch >
A sketch matrix storing sketches for each feature.