#include <arima_common.h>
Collaboration diagram for ML::ARIMAParams< DataT >:
Public Member Functions | |
| void | allocate (const ARIMAOrder &order, int batch_size, cudaStream_t stream, bool tr=false) |
| void | deallocate (const ARIMAOrder &order, int batch_size, cudaStream_t stream, bool tr=false) |
| void | pack (const ARIMAOrder &order, int batch_size, DataT *param_vec, cudaStream_t stream) const |
| void | unpack (const ARIMAOrder &order, int batch_size, const DataT *param_vec, cudaStream_t stream) |
Public Attributes | |
| DataT * | mu = nullptr |
| DataT * | beta = nullptr |
| DataT * | ar = nullptr |
| DataT * | ma = nullptr |
| DataT * | sar = nullptr |
| DataT * | sma = nullptr |
| DataT * | sigma2 = nullptr |
Detailed Description
template<typename DataT>
struct ML::ARIMAParams< DataT >
Structure to hold the parameters (makes it easier to pass as an argument)
- Note
- : the qualifier const applied to this structure will only guarantee that the pointers are not changed, but the user can still modify the arrays when using the pointers directly!
Member Function Documentation
◆ allocate()
template<typename DataT >
|
inline |
Allocate all the parameter device arrays
- Template Parameters
-
AllocatorT Type of allocator used
- Parameters
-
[in] order ARIMA order [in] batch_size Batch size [in] stream CUDA stream [in] tr Whether these are the transformed parameters
◆ deallocate()
template<typename DataT >
|
inline |
Deallocate all the parameter device arrays
- Template Parameters
-
AllocatorT Type of allocator used
- Parameters
-
[in] order ARIMA order [in] batch_size Batch size [in] stream CUDA stream [in] tr Whether these are the transformed parameters
◆ pack()
template<typename DataT >
|
inline |
Pack the separate parameter arrays into a unique parameter vector
- Parameters
-
[in] order ARIMA order [in] batch_size Batch size [out] param_vec Linear array of all parameters grouped by batch [mu, ar, ma, sar, sma, sigma2] (device) [in] stream CUDA stream
◆ unpack()
template<typename DataT >
|
inline |
Unpack a parameter vector into separate arrays of parameters.
- Parameters
-
[in] order ARIMA order [in] batch_size Batch size [in] param_vec Linear array of all parameters grouped by batch [mu, ar, ma, sar, sma, sigma2] (device) [in] stream CUDA stream
Member Data Documentation
◆ ar
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::ar = nullptr |
◆ beta
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::beta = nullptr |
◆ ma
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::ma = nullptr |
◆ mu
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::mu = nullptr |
◆ sar
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::sar = nullptr |
◆ sigma2
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::sigma2 = nullptr |
◆ sma
template<typename DataT >
| DataT* ML::ARIMAParams< DataT >::sma = nullptr |
The documentation for this struct was generated from the following file:
- include/cuml/tsa/arima_common.h
