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

View File

@@ -40,16 +40,25 @@ class Input : public SequentialLayer {
*/
int getOutputSize();
/**
/**
* @brief Get input size
*
*
* @return int input size
*/
int getInputSize();
private:
int inputSize;
int inputSize;
float* forwardCPU(const float* input);
#ifdef USE_CUDA
float* d_output;
float* forwardCUDA(const float* input);
void initCUDA();
void delCUDA();
#endif
};
} // namespace CUDANet::Layers