|
nnlib
GPU-accelerated, C/C++ neural network library.
|
| ▼ examples | |
| ▼ mnist | |
| main.cpp | Source file containing an example usage of the library for digit recognition |
| ▼ titanic | |
| main.cpp | Source file containing an example usage of the library for the Titanic dataset |
| ▼ include | |
| allocation.h | Header file to declare common functions regarding memory allocation on host |
| cache.h | |
| functions.h | |
| layer.h | Header file declaring the Layer class |
| loss.h | Header file declaring different loss functions |
| metric.h | Header file declaring different metrics |
| network.h | Header file declaring the Network class |
| onehot_encode.h | Header file containing the declaration of the oneHotEncode() method |
| read.h | Header file declaring methods that can be used when reading from files |
| runtime.h | |
| session.cuh | Header file declaring the Session class |
| tensor.h | Header file declaring the Tensor class to represent multidimensional arrays |
| tuple_utils.h | |
| verify.cuh | Header file declaring methods used for verifying the existence of a CUDA runtime |
| ▼ src | |
| ▼ ann | |
| ▼ losses | |
| binary_cross_entropy.cpp | Source file defining the methods of the BinaryCrossEntropy class |
| categorical_cross_entropy.cpp | Source file defining the methods of the CategoricalCrossEntropy class |
| loss.cpp | Source file defining the methods of the Loss class |
| mean_squared_error.cpp | Source file defining the methods of the MeanSquaredError class |
| ▼ metrics | |
| binary_accuracy.cpp | Source file defining the methods of the BinaryAccuracy class |
| categorical_accuracy.cpp | Source file defining the methods of the CategoricalAccuracy class |
| metric.cpp | Source file defining the methods of the Metric class |
| layer.cpp | Source file defining methods of the Layer class |
| network.cpp | Source file defining methods of the Network class |
| ▼ exceptions | |
| different_data_location_exception.cpp | Source file defining the DifferentDataLocationException class |
| different_data_location_exception.h | Header file declaring the DifferentDataLocationException class |
| size_mismatch_exception.cpp | Source file defining the SizeMismatchException class |
| size_mismatch_exception.h | Header file declaring the SizeMismatchException class |
| unexpected_cuda_call_exception.cpp | Source file defining the UnexpectedCUDACallException class |
| unexpected_cuda_call_exception.h | Header file declaring the UnexpectedCUDACallException class |
| unsupported_operation_exception.cpp | Source file defining the UnsupportedOperationException class |
| unsupported_operation_exception.h | Header file declaring the UnsupportedOperationException class |
| ▼ gpu | |
| allocation_gpu.cu | Source file defining methods regarding memory allocation on device |
| allocation_gpu.cuh | Header file to declare common functions regarding memory allocation on device |
| assert.cuh | Header file declaring the gpuAssert() method |
| verify.cu | Source file defining methods used for verifying the existence of a CUDA runtime |
| ▼ math | |
| tensor.cpp | Source file defining methods related to the Tensor class |
| tensor_operations_on_device.cu | Source file defining tensor operations that happen on device |
| tensor_operations_on_device.cuh | Header file declaring tensor operations that happen on device |
| tensor_operations_on_host.cpp | Source file defining tensor operations that happen on host |
| tensor_operations_on_host.h | Header file declaring tensor operations that happen on host |
| ▼ runtime | |
| cache.cpp | |
| runtime.cpp | |
| ▼ utils | |
| allocation.cpp | Source file defining methods regarding memory allocation on host |
| location_verifiers.cpp | Header file defining methods to verify multiple DataLocation at once |
| location_verifiers.h | Header file declaring methods to verify multiple DataLocation at once |
| onehot_encode.cpp | Source file containing the definition of the oneHotEncode() method |
| printing.cpp | Source file defining methods to construct useful string messages |
| printing.h | Header file declaring methods to construct useful string messages |
| read.cpp | Source file defining methods that can be used when reading from files |
| session.cu | Source file defining the Session class |