mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Migrate Activation layer
This commit is contained in:
@@ -47,11 +47,21 @@ class Activation {
|
||||
private:
|
||||
ActivationType activationType;
|
||||
int length;
|
||||
|
||||
#ifdef USE_CUDA
|
||||
int gridSize;
|
||||
|
||||
float* d_softmax_sum;
|
||||
float* d_max;
|
||||
|
||||
void activateCUDA(float* d_input);
|
||||
|
||||
void initCUDA();
|
||||
void delCUDA();
|
||||
#else
|
||||
void activateCPU(float* input);
|
||||
#endif
|
||||
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user