nnlib
GPU-accelerated, C/C++ neural network library.
Public Member Functions | Static Public Member Functions | Private Attributes | List of all members
Cache Class Reference

Public Member Functions

float * get (size_t size, DataLocation location)
 
void put (size_t size, float *ptr, DataLocation location)
 
 Cache (Cache const &)=delete
 
void operator= (Cache const &)=delete
 

Static Public Member Functions

static CachegetInstance ()
 

Private Attributes

std::unordered_map< size_t, std::stack< float * > > hostCache
 
std::unordered_map< size_t, std::stack< float * > > deviceCache
 

The documentation for this class was generated from the following files: