A container for managed prediction caches. More...
#include <predictor.h>
Inheritance diagram for xgboost::PredictionContainer:
Public Member Functions | |
| PredictionCacheEntry & | Cache (std::shared_ptr< DMatrix > m, std::int32_t device) |
 Public Member Functions inherited from xgboost::DMatrixCache< PredictionCacheEntry > | |
| DMatrixCache (std::size_t cache_size) | |
| DMatrixCache & | operator= (DMatrixCache &&that) |
| std::shared_ptr< PredictionCacheEntry > | CacheItem (std::shared_ptr< DMatrix > m, Args const &... args) |
| Cache a new DMatrix if it's not in the cache already. | |
| std::shared_ptr< PredictionCacheEntry > | ResetItem (std::shared_ptr< DMatrix > m, Args const &... args) |
| Re-initialize the item in cache. | |
| decltype(container_) const & | Container () |
| Get a const reference to the underlying hash map. | |
| std::shared_ptr< PredictionCacheEntry > | Entry (DMatrix const *m) const |
Additional Inherited Members | |
| Static Public Member Functions inherited from xgboost::DMatrixCache< PredictionCacheEntry > | |
| static constexpr std::size_t | DefaultSize () |
| Protected Member Functions inherited from xgboost::DMatrixCache< PredictionCacheEntry > | |
| void | CheckConsistent () const |
| void | ClearExpired () |
| void | ClearExcess () |
| Protected Attributes inherited from xgboost::DMatrixCache< PredictionCacheEntry > | |
| std::unordered_map< Key, Item, Hash > | container_ |
| std::queue< Key > | queue_ |
| std::size_t | max_size_ |
Detailed Description
A container for managed prediction caches.
The documentation for this class was generated from the following file:
- External/xgboost/include/xgboost/predictor.h
