xgboost
|
Implementation of gradient statistics pair. Template specialisation may be used to overload different gradients types e.g. low precision, high precision, integer, floating point. More...
#include <base.h>
Public Types | |
using | ValueT = T |
Public Member Functions | |
XGBOOST_DEVICE | GradientPairInternal () |
XGBOOST_DEVICE | GradientPairInternal (float grad, float hess) |
XGBOOST_DEVICE | GradientPairInternal (const GradientPairInternal< T > &g) |
template<typename T2 > | |
XGBOOST_DEVICE | GradientPairInternal (const GradientPairInternal< T2 > &g) |
XGBOOST_DEVICE float | GetGrad () const |
XGBOOST_DEVICE float | GetHess () const |
XGBOOST_DEVICE GradientPairInternal< T > & | operator+= (const GradientPairInternal< T > &rhs) |
XGBOOST_DEVICE GradientPairInternal< T > | operator+ (const GradientPairInternal< T > &rhs) const |
XGBOOST_DEVICE GradientPairInternal< T > & | operator-= (const GradientPairInternal< T > &rhs) |
XGBOOST_DEVICE GradientPairInternal< T > | operator- (const GradientPairInternal< T > &rhs) const |
XGBOOST_DEVICE | GradientPairInternal (int value) |
template<> | |
XGBOOST_DEVICE float | GetGrad () const |
template<> | |
XGBOOST_DEVICE float | GetHess () const |
Friends | |
std::ostream & | operator<< (std::ostream &os, const GradientPairInternal< T > &g) |
Implementation of gradient statistics pair. Template specialisation may be used to overload different gradients types e.g. low precision, high precision, integer, floating point.
using xgboost::detail::GradientPairInternal< T >::ValueT = T |
|
inline |
|
inline |
|
inline |
|
inlineexplicit |
|
inlineexplicit |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
friend |