mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
16 lines
354 B
Plaintext
16 lines
354 B
Plaintext
#ifndef CONVOLUTION_H
|
|
#define CONVOLUTION_H
|
|
|
|
__global__ void convolution_kernel(
|
|
const float* d_input,
|
|
const float* d_kernel,
|
|
float* d_output,
|
|
int inputSize,
|
|
int nChannels,
|
|
int kernelSize,
|
|
int stride,
|
|
int nFilters,
|
|
int outputSize
|
|
);
|
|
|
|
#endif // CONVOLUTION_H |