Public Member Functions
	CommonRowPartitioner (Context const *ctx, bst_row_t num_row, bst_row_t _base_rowid, bool is_col_split)

template<typename ExpandEntry >
void	FindSplitConditions (const std::vector< ExpandEntry > &nodes, const RegTree &tree, const GHistIndexMatrix &gmat, std::vector< int32_t > *split_conditions)

template<typename ExpandEntry >
void	AddSplitsToRowSet (const std::vector< ExpandEntry > &nodes, RegTree const *p_tree)

template<typename ExpandEntry >
void	UpdatePosition (Context const ctx, GHistIndexMatrix const &gmat, std::vector< ExpandEntry > const &nodes, RegTree const p_tree)

template<bool any_cat, typename ExpandEntry >
void	UpdatePosition (Context const ctx, GHistIndexMatrix const &gmat, const common::ColumnMatrix &column_matrix, std::vector< ExpandEntry > const &nodes, RegTree const p_tree)

template<bool any_missing, bool any_cat, typename ExpandEntry >
void	UpdatePosition (Context const ctx, GHistIndexMatrix const &gmat, const common::ColumnMatrix &column_matrix, std::vector< ExpandEntry > const &nodes, RegTree const p_tree)

template<typename BinIdxType , bool any_missing, bool any_cat, typename ExpandEntry >
void	UpdatePosition (Context const ctx, GHistIndexMatrix const &gmat, const common::ColumnMatrix &column_matrix, std::vector< ExpandEntry > const &nodes, RegTree const p_tree)

auto const &	Partitions () const

std::size_t	Size () const

auto &	operator[] (bst_node_t nidx)

auto const &	operator[] (bst_node_t nidx) const

void	LeafPartition (Context const ctx, RegTree const &tree, common::Span< float const > hess, std::vector< bst_node_t > p_out_position) const

void	LeafPartition (Context const ctx, RegTree const &tree, linalg::TensorView< GradientPair const, 2 > gpair, std::vector< bst_node_t > p_out_position) const

void	LeafPartition (Context const ctx, RegTree const &tree, common::Span< GradientPair const > gpair, std::vector< bst_node_t > p_out_position) const

Data Fields
bst_row_t	base_rowid = 0

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

External/xgboost/src/tree/common_row_partitioner.h

Public Member Functions

Data Fields