Cleanup and refactor

This commit is contained in:
2024-04-11 22:52:41 +02:00
parent 4b9d123e94
commit 18522c2dea
9 changed files with 81 additions and 72 deletions

View File

@@ -28,7 +28,7 @@ class Activation {
* @param activation Type of activation
* @param length Length of the input
*/
Activation(ActivationType activation, const unsigned int length);
Activation(ActivationType activation, const int length);
/**
* @brief Destroy the Activation object
@@ -46,8 +46,8 @@ class Activation {
private:
ActivationType activationType;
unsigned int length;
unsigned int gridSize;
int length;
int gridSize;
float* d_softmax_sum;
float* d_max;