mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Restructure include paths
This commit is contained in:
28
include/backend/cuda/kernels/pool.cuh
Normal file
28
include/backend/cuda/kernels/pool.cuh
Normal file
@@ -0,0 +1,28 @@
|
||||
#pragma once
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
#include "layer.hpp"
|
||||
|
||||
namespace CUDANet::Kernels {
|
||||
|
||||
__global__ void max_pool(
|
||||
const float* __restrict__ d_input,
|
||||
float* __restrict__ d_output,
|
||||
const Shape input_shape,
|
||||
const Shape output_shape,
|
||||
const Shape pool_shape,
|
||||
const Shape stride_shape,
|
||||
const Shape padding_shape
|
||||
);
|
||||
|
||||
__global__ void avg_pool(
|
||||
const float* __restrict__ d_input,
|
||||
float* __restrict__ d_output,
|
||||
const Shape input_shape,
|
||||
const Shape output_shape,
|
||||
const Shape pool_shape,
|
||||
const Shape stride_shape,
|
||||
const Shape padding_shape
|
||||
);
|
||||
|
||||
} // namespace CUDANet::Kernels
|
||||
Reference in New Issue
Block a user