Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Data Fields | Friends
xgboost::tree::MultiExpandEntry Struct Reference
Inheritance diagram for xgboost::tree::MultiExpandEntry:
xgboost::tree::ExpandEntryImpl< MultiExpandEntry >

Public Member Functions

 MultiExpandEntry (bst_node_t nidx, bst_node_t depth)
 
bool IsValidImpl (TrainParam const &param, bst_node_t num_leaves) const
 
void CopyAndCollect (MultiExpandEntry const &that, std::vector< uint32_t > *collected_cat_bits, std::vector< std::size_t > *cat_bits_sizes, std::vector< GradientPairPrecise > *collected_gradients)
 Copy primitive fields into this, and collect cat_bits and gradients into vectors.
 
- Public Member Functions inherited from xgboost::tree::ExpandEntryImpl< MultiExpandEntry >
float GetLossChange () const
 
bst_node_t GetNodeId () const
 
bool IsValid (TrainParam const &param, bst_node_t num_leaves) const
 

Data Fields

SplitEntryContainer< std::vector< GradientPairPrecise > > split
 
- Data Fields inherited from xgboost::tree::ExpandEntryImpl< MultiExpandEntry >
bst_node_t nid
 
bst_node_t depth
 

Friends

std::ostream & operator<< (std::ostream &os, MultiExpandEntry const &e)
 

Member Function Documentation

◆ CopyAndCollect()

void xgboost::tree::MultiExpandEntry::CopyAndCollect ( MultiExpandEntry const &  that,
std::vector< uint32_t > *  collected_cat_bits,
std::vector< std::size_t > *  cat_bits_sizes,
std::vector< GradientPairPrecise > *  collected_gradients 
)
inline

Copy primitive fields into this, and collect cat_bits and gradients into vectors.

This is used for allgather.

Parameters
thatThe other entry to copy from
collected_cat_bitsThe vector to collect cat_bits
cat_bits_sizesThe sizes of the collected cat_bits
collected_gradientsThe vector to collect gradients

The documentation for this struct was generated from the following file: