mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Add Kernels namespace
This commit is contained in:
@@ -1,7 +1,18 @@
|
||||
#ifndef CONVOLUTION_H
|
||||
#define CONVOLUTION_H
|
||||
|
||||
__global__ void convolution_kernel(
|
||||
namespace Kernels {
|
||||
|
||||
__global__ void padding(
|
||||
const float* d_input,
|
||||
float* d_padded,
|
||||
int w,
|
||||
int h,
|
||||
int n,
|
||||
int p
|
||||
);
|
||||
|
||||
__global__ void convolution(
|
||||
const float* d_input,
|
||||
const float* d_kernel,
|
||||
float* d_output,
|
||||
@@ -13,4 +24,6 @@ __global__ void convolution_kernel(
|
||||
int outputSize
|
||||
);
|
||||
|
||||
} // namespace Kernels
|
||||
|
||||
#endif // CONVOLUTION_H
|
||||
Reference in New Issue
Block a user