#include <cuml/experimental/fil/detail/index_type.hpp>#include <cuml/experimental/fil/detail/infer/cpu.hpp>#include <cuml/experimental/fil/detail/postprocessor.hpp>#include <cuml/experimental/fil/detail/raft_proto/cuda_stream.hpp>#include <cuml/experimental/fil/detail/raft_proto/device_id.hpp>#include <cuml/experimental/fil/detail/raft_proto/device_type.hpp>#include <cuml/experimental/fil/exceptions.hpp>#include <cuml/experimental/fil/infer_kind.hpp>#include <cstddef>#include <iostream>#include <optional>#include <type_traits>
Include dependency graph for infer.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<raft_proto::device_type D, typename forest_t > | |
| void | ML::experimental::fil::detail::infer (forest_t const &forest, postprocessor< typename forest_t::io_type > const &postproc, typename forest_t::io_type *output, typename forest_t::io_type *input, index_type row_count, index_type col_count, index_type output_count, bool has_categorical_nodes, typename forest_t::io_type *vector_output=nullptr, typename forest_t::node_type::index_type *categorical_data=nullptr, infer_kind infer_type=infer_kind::default_kind, std::optional< index_type > specified_chunk_size=std::nullopt, raft_proto::device_id< D > device=raft_proto::device_id< D >{}, raft_proto::cuda_stream stream=raft_proto::cuda_stream{}) |
