ML::experimental::fil::detail::device_initialization Namespace Reference

Functions
template<typename forest_t , raft_proto::device_type D>
std::enable_if_t< std::disjunction_v< std::bool_constant<!raft_proto::GPU_ENABLED >, std::bool_constant< D==raft_proto::device_type::cpu > >, void >	initialize_device (raft_proto::device_id< D > device)
template<typename forest_t , raft_proto::device_type D>
std::enable_if_t< std::conjunction_v< std::bool_constant< raft_proto::GPU_ENABLED >, std::bool_constant< D==raft_proto::device_type::gpu > >, void >	initialize_device (raft_proto::device_id< D > device)

Function Documentation

initialize_device() [1/2]

template<typename forest_t , raft_proto::device_type D>

std::enable_if_t<std::disjunction_v<std::bool_constant<!raft_proto::GPU_ENABLED>, std::bool_constant<D == raft_proto::device_type::cpu> >, void> ML::experimental::fil::detail::device_initialization::initialize_device

(

raft_proto::device_id< D >

device

)

initialize_device() [2/2]

template<typename forest_t , raft_proto::device_type D>

std::enable_if_t<std::conjunction_v<std::bool_constant<raft_proto::GPU_ENABLED>, std::bool_constant<D == raft_proto::device_type::gpu> >, void> ML::experimental::fil::detail::device_initialization::initialize_device

(

raft_proto::device_id< D >

device

)