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)