#include <type_traits>
#include "common.h"
#include "threading_utils.h"
#include "xgboost/generic_parameters.h"
#include "xgboost/linalg.h"
Go to the source code of this file.
|
template<typename T , int32_t D, typename Fn > |
void | xgboost::linalg::ElementWiseTransformHost (linalg::TensorView< T, D > t, int32_t n_threads, Fn &&fn) |
|
template<typename T , int32_t D, typename Fn > |
void | xgboost::linalg::ElementWiseKernelHost (linalg::TensorView< T, D > t, int32_t n_threads, Fn &&fn) |
|
template<typename T , int32_t D, typename Fn > |
void | xgboost::linalg::ElementWiseKernelDevice (linalg::TensorView< T, D > t, Fn &&fn, void *s=nullptr) |
|
template<typename T , int32_t D, typename Fn > |
void | xgboost::linalg::ElementWiseTransformDevice (linalg::TensorView< T, D > t, Fn &&fn, void *s=nullptr) |
|
template<typename T , int32_t D, typename Fn > |
void | xgboost::linalg::ElementWiseKernel (GenericParameter const *ctx, linalg::TensorView< T, D > t, Fn &&fn) |
|