Migrate input layer

This commit is contained in:
2024-09-10 19:11:21 +02:00
parent fe7c16ac36
commit f7b525e494
7 changed files with 57 additions and 20 deletions

37
src/layers/input.cpp Normal file
View File

@@ -0,0 +1,37 @@
#include <stdexcept>
#include "input.hpp"
using namespace CUDANet::Layers;
Input::Input(int inputSize) : inputSize(inputSize) {
#ifdef USE_CUDA
initCUDA();
#endif
}
Input::~Input() {
#ifdef USE_CUDA
delCUDA();
#endif
}
float* Input::forwardCPU(const float* input) {
throw std::logic_error("Not implemented");
}
float* Input::forward(const float* input) {
#ifdef USE_CUDA
return forwardCUDA(input);
#else
return forwardCPU(input);
#endif
}
int Input::getOutputSize() {
return inputSize;
}
int Input::getInputSize() {
return inputSize;
}