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

Migrate output layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-10 19:20:00 +02:00
parent f7b525e494
commit 74f49d6a00
6 changed files with 34 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/layers/output.cuh → include/layers/output.hpp
View File

@@ -46,6 +46,12 @@ class Output : public SequentialLayer {
private:
int inputSize;
float* h_output;
float* forwardCPU(const float* input);
#ifdef USE_CUDA
float* forwardCUDA(const float* input);
#endif
};
} // namespace CUDANet::Layers