Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Migrate avg pooling

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-09 21:36:13 +02:00
parent 75475790ac
commit 76e5225001
5 changed files with 129 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
include/cudanet.cuh
View File

@@ -1,16 +1,17 @@
#ifndef CUDANET_H
#define CUDANET_H
// Kernels
#ifdef USE_CUDA
#include "activation_functions.cuh"
#include "convolution.cuh"
#include "matmul.cuh"
#include "pooling.cuh"
#endif
// Layers
#include "activation.hpp"
#include "add.cuh"
#include "avg_pooling.cuh"
#include "add.hpp"
#include "avg_pooling.hpp"
#include "batch_norm.cuh"
#include "concat.cuh"
#include "conv2d.cuh"
@@ -25,8 +26,10 @@
#include "module.hpp"
// Utils
#include "cuda_helper.cuh"
#include "imagenet.hpp"
#ifdef USE_CUDA
#include "cuda_helper.cuh"
#include "vector.cuh"
#endif
#endif // CUDANET_H