mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-07 02:04:26 +00:00
Add toplevel CUDANet namespace
This commit is contained in:
@@ -7,7 +7,7 @@
|
||||
|
||||
#include "ilayer.cuh"
|
||||
|
||||
namespace Layers {
|
||||
namespace CUDANet::Layers {
|
||||
|
||||
/**
|
||||
* @brief Dense (fully connected) layer
|
||||
@@ -53,8 +53,8 @@ class Dense : public ILayer {
|
||||
void setBiases(const float* biases);
|
||||
|
||||
private:
|
||||
int inputSize;
|
||||
int outputSize;
|
||||
unsigned int inputSize;
|
||||
unsigned int outputSize;
|
||||
|
||||
float* d_output;
|
||||
|
||||
@@ -67,8 +67,8 @@ class Dense : public ILayer {
|
||||
Layers::Activation activation;
|
||||
|
||||
// Precompute kernel launch parameters
|
||||
int forwardGridSize;
|
||||
int biasGridSize;
|
||||
unsigned int forwardGridSize;
|
||||
unsigned int biasGridSize;
|
||||
|
||||
/**
|
||||
* @brief Initialize the weights to zeros
|
||||
@@ -89,6 +89,6 @@ class Dense : public ILayer {
|
||||
void toCuda();
|
||||
};
|
||||
|
||||
} // namespace Layers
|
||||
} // namespace CUDANet::Layers
|
||||
|
||||
#endif // CUDANET_DENSE_LAYER_H
|
||||
|
||||
Reference in New Issue
Block a user