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

Public Member Functions

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

Data Fields

SplitEntry split
 
- Data Fields inherited from xgboost::tree::ExpandEntryImpl< CPUExpandEntry >
bst_node_t nid
 
bst_node_t depth
 

Friends

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

Member Function Documentation

◆ CopyAndCollect()

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

Copy primitive fields into this, and collect cat_bits into a vector.

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

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