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)