#include <partition_builder.h>
|
template<typename Func > |
void | Init (const size_t n_tasks, size_t n_nodes, Func funcNTask) |
|
template<bool default_left, bool any_missing, typename ColumnType > |
std::pair< size_t, size_t > | PartitionKernel (const ColumnType &column, common::Span< const size_t > rid_span, const int32_t split_cond, common::Span< size_t > left_part, common::Span< size_t > right_part) |
|
template<typename BinIdxType , bool any_missing> |
void | Partition (const size_t node_in_set, const size_t nid, const common::Range1d range, const int32_t split_cond, const ColumnMatrix &column_matrix, const RegTree &tree, const size_t *rid) |
|
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) |
|
◆ AllocateForTask()
template<size_t BlockSize>
◆ CalculateRowOffsets()
template<size_t BlockSize>
◆ GetLeftBuffer()
template<size_t BlockSize>
◆ GetNLeftElems()
template<size_t BlockSize>
◆ GetNRightElems()
template<size_t BlockSize>
◆ GetRightBuffer()
template<size_t BlockSize>
◆ GetTaskIdx()
template<size_t BlockSize>
◆ Init()
template<size_t BlockSize>
template<typename Func >
◆ MergeToArray()
template<size_t BlockSize>
◆ Partition()
template<size_t BlockSize>
template<typename BinIdxType , bool any_missing>
◆ PartitionKernel()
template<size_t BlockSize>
template<bool default_left, bool any_missing, typename ColumnType >
◆ SetNLeftElems()
template<size_t BlockSize>
◆ SetNRightElems()
template<size_t BlockSize>
◆ blocks_offsets_
template<size_t BlockSize>
◆ left_right_nodes_sizes_
template<size_t BlockSize>
◆ max_n_tasks_
template<size_t BlockSize>
◆ mem_blocks_
template<size_t BlockSize>
The documentation for this class was generated from the following file: