Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-19 21:35:05 +01:00
parent 8d14b74f66
commit b6c4b7d2ae
12 changed files with 87 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/layers/add.cuh
View File

@@ -19,14 +19,13 @@ class Add {
~Add();
/**
* @brief Adds the two inputs
* @brief Adds first input to second input
*
* @param d_inputA Device pointer to the first input
* @param d_inputB Device pointer to the second input
*
* @return Device pointer to the output
*/
float* forward(const float* d_inputA, const float* d_inputB);
void forward(const float* d_inputA, const float* d_inputB);
private:
int inputSize;