Namespaces | Functions
evaluate_tree.hpp File Reference
#include <stdint.h>
#include <type_traits>
#include <math.h>
#include <cuml/experimental/fil/detail/bitset.hpp>
#include <cuml/experimental/fil/detail/raft_proto/gpu_support.hpp>
Include dependency graph for evaluate_tree.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 ML
 
 ML::experimental
 
 ML::experimental::fil
 
 ML::experimental::fil::detail
 

Functions

template<bool has_vector_leaves, bool has_categorical_nodes, typename node_t , typename io_t , typename node_id_mapping_t = std::nullptr_t>
HOST DEVICE auto ML::experimental::fil::detail::evaluate_tree_impl (node_t const *__restrict__ node, io_t const *__restrict__ row, node_t const *__restrict__ first_root_node=nullptr, node_id_mapping_t node_id_mapping=nullptr)
 
template<bool has_vector_leaves, typename node_t , typename io_t , typename categorical_storage_t , typename node_id_mapping_t = std::nullptr_t>
HOST DEVICE auto ML::experimental::fil::detail::evaluate_tree_impl (node_t const *__restrict__ node, io_t const *__restrict__ row, categorical_storage_t const *__restrict__ categorical_storage, node_t const *__restrict__ first_root_node=nullptr, node_id_mapping_t node_id_mapping=nullptr)
 
template<bool has_vector_leaves, bool has_categorical_nodes, bool has_nonlocal_categories, bool predict_leaf, typename forest_t , typename io_t , typename categorical_data_t >
HOST DEVICE auto ML::experimental::fil::detail::evaluate_tree (forest_t const &forest, index_type tree_index, io_t const *__restrict__ row, categorical_data_t categorical_data)