xgboost::common::PartitionBuilder< BlockSize > Class Template Reference

#include <row_set.h>

Collaboration diagram for xgboost::common::PartitionBuilder< BlockSize >:

Classes
struct	BlockInfo

Public Member Functions
template<typename Func >
void	Init (const size_t n_tasks, size_t n_nodes, Func funcNTaks)
void	AllocateForTask (size_t id)
common::Span< size_t >	GetLeftBuffer (int nid, size_t begin, size_t end)
common::Span< size_t >	GetRightBuffer (int nid, size_t begin, size_t end)
void	SetNLeftElems (int nid, size_t begin, size_t end, size_t n_left)
void	SetNRightElems (int nid, size_t begin, size_t end, size_t n_right)
size_t	GetNLeftElems (int nid) const
size_t	GetNRightElems (int nid) const
void	CalculateRowOffsets ()
void	MergeToArray (int nid, size_t begin, size_t *rows_indexes)
size_t	GetTaskIdx (int nid, size_t begin)

Protected Attributes
std::vector< std::pair< size_t, size_t > >	left_right_nodes_sizes_
std::vector< size_t >	blocks_offsets_
std::vector< std::shared_ptr< BlockInfo > >	mem_blocks_
size_t	max_n_tasks_ = 0

Member Function Documentation

AllocateForTask()

template<size_t BlockSize>

void xgboost::common::PartitionBuilder< BlockSize >::AllocateForTask ( size_t  id )

inline

CalculateRowOffsets()

template<size_t BlockSize>

void xgboost::common::PartitionBuilder< BlockSize >::CalculateRowOffsets ( )

inline

GetLeftBuffer()

template<size_t BlockSize>

common::Span<size_t> xgboost::common::PartitionBuilder< BlockSize >::GetLeftBuffer ( int  nid, size_t  begin, size_t  end  )

inline

GetNLeftElems()

template<size_t BlockSize>

size_t xgboost::common::PartitionBuilder< BlockSize >::GetNLeftElems ( int  nid ) const

inline

GetNRightElems()

template<size_t BlockSize>

size_t xgboost::common::PartitionBuilder< BlockSize >::GetNRightElems ( int  nid ) const

inline

GetRightBuffer()

template<size_t BlockSize>

common::Span<size_t> xgboost::common::PartitionBuilder< BlockSize >::GetRightBuffer ( int  nid, size_t  begin, size_t  end  )

inline

GetTaskIdx()

template<size_t BlockSize>

size_t xgboost::common::PartitionBuilder< BlockSize >::GetTaskIdx ( int  nid, size_t  begin  )

inline

Init()

template<size_t BlockSize>

template<typename Func >

void xgboost::common::PartitionBuilder< BlockSize >::Init ( const size_t  n_tasks, size_t  n_nodes, Func  funcNTaks  )

inline

MergeToArray()

template<size_t BlockSize>

void xgboost::common::PartitionBuilder< BlockSize >::MergeToArray ( int  nid, size_t  begin, size_t *  rows_indexes  )

inline

SetNLeftElems()

template<size_t BlockSize>

void xgboost::common::PartitionBuilder< BlockSize >::SetNLeftElems ( int  nid, size_t  begin, size_t  end, size_t  n_left  )

inline

SetNRightElems()

template<size_t BlockSize>

void xgboost::common::PartitionBuilder< BlockSize >::SetNRightElems ( int  nid, size_t  begin, size_t  end, size_t  n_right  )

inline

Member Data Documentation

blocks_offsets_

template<size_t BlockSize>

std::vector<size_t> xgboost::common::PartitionBuilder< BlockSize >::blocks_offsets_

protected

left_right_nodes_sizes_

template<size_t BlockSize>

std::vector<std::pair<size_t, size_t> > xgboost::common::PartitionBuilder< BlockSize >::left_right_nodes_sizes_

protected

max_n_tasks_

template<size_t BlockSize>

size_t xgboost::common::PartitionBuilder< BlockSize >::max_n_tasks_ = 0

protected

mem_blocks_

template<size_t BlockSize>

std::vector<std::shared_ptr<BlockInfo> > xgboost::common::PartitionBuilder< BlockSize >::mem_blocks_

protected

---

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

• /home/runner/work/xgboost/xgboost/src/common/row_set.h