|
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 |