nnlib
GPU-accelerated, C/C++ neural network library.
|
#include <cstdlib>
#include <stack>
#include <unordered_map>
Go to the source code of this file.
Classes | |
class | Cache |
Enumerations | |
enum | DataLocation { HOST , DEVICE } |
Enumerate to specify where data is located. More... | |
enum DataLocation |
Enumerate to specify where data is located.
Can be either HOST or DEVICE. In case it is set to HOST, the data is stored in RAM and is processed by the CPU. In case it is set to DEVICE, the data is in VRAM and processed by the GPU. The latter is only possible if CUDA is installed and there is a CUDA enabled GPU on the system.