nnlib
GPU-accelerated, C/C++ neural network library.
|
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 Cache & | getInstance () |
Private Attributes | |
std::unordered_map< size_t, std::stack< float * > > | hostCache |
std::unordered_map< size_t, std::stack< float * > > | deviceCache |