xgboost
Namespaces | Functions
linalg_op.h File Reference
#include <type_traits>
#include "common.h"
#include "threading_utils.h"
#include "xgboost/generic_parameters.h"
#include "xgboost/linalg.h"
Include dependency graph for linalg_op.h:

Go to the source code of this file.

Namespaces

 xgboost
 namespace of xgboost
 
 xgboost::linalg
 

Functions

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)