mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
Add main include file
This commit is contained in:
32
include/cudanet.cuh
Normal file
32
include/cudanet.cuh
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
#ifndef CUDANET_H
|
||||||
|
#define CUDANET_H
|
||||||
|
|
||||||
|
// Kernels
|
||||||
|
#include "activation_functions.cuh"
|
||||||
|
#include "convolution.cuh"
|
||||||
|
#include "matmul.cuh"
|
||||||
|
#include "pooling.cuh"
|
||||||
|
|
||||||
|
// Layers
|
||||||
|
#include "activation.cuh"
|
||||||
|
#include "add.cuh"
|
||||||
|
#include "avg_pooling.cuh"
|
||||||
|
#include "batch_norm.cuh"
|
||||||
|
#include "concat.cuh"
|
||||||
|
#include "conv2d.cuh"
|
||||||
|
#include "dense.cuh"
|
||||||
|
#include "input.cuh"
|
||||||
|
#include "layer.cuh"
|
||||||
|
#include "max_pooling.cuh"
|
||||||
|
#include "output.cuh"
|
||||||
|
|
||||||
|
// Models
|
||||||
|
#include "model.hpp"
|
||||||
|
#include "module.hpp"
|
||||||
|
|
||||||
|
// Utils
|
||||||
|
#include "cuda_helper.cuh"
|
||||||
|
#include "imagenet.hpp"
|
||||||
|
#include "vector.cuh"
|
||||||
|
|
||||||
|
#endif // CUDANET_H
|
||||||
Reference in New Issue
Block a user