mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
56 lines
1.5 KiB
C++
56 lines
1.5 KiB
C++
#pragma once
|
|
|
|
// ============================================================================
|
|
// Core Data Structures & Abstractions (BACKEND-INDEPENDENT)
|
|
// ============================================================================
|
|
|
|
#include "shape.hpp"
|
|
#include "backend.hpp"
|
|
#include "tensor.hpp"
|
|
#include "layer.hpp"
|
|
|
|
// ============================================================================
|
|
// Container Classes
|
|
// ============================================================================
|
|
|
|
#include "module.hpp"
|
|
#include "model.hpp"
|
|
|
|
// ============================================================================
|
|
// Layer Implementations
|
|
// ============================================================================
|
|
|
|
// Activation
|
|
#include "layers/activation.hpp"
|
|
|
|
// Normalization
|
|
#include "layers/batch_norm.hpp"
|
|
|
|
// Linear
|
|
#include "layers/dense.hpp"
|
|
|
|
// Convolutional
|
|
#include "layers/conv2d.hpp"
|
|
|
|
// Pooling
|
|
#include "layers/max_pool.hpp"
|
|
#include "layers/avg_pool.hpp"
|
|
|
|
// Composition (element-wise operations)
|
|
#include "layers/add.hpp"
|
|
#include "layers/concat.hpp"
|
|
|
|
// ============================================================================
|
|
// Dataset Labels
|
|
// ============================================================================
|
|
|
|
#include "datasets/imagenet.hpp"
|
|
|
|
// ============================================================================
|
|
// Backend-Specific Includes (conditionally compiled)
|
|
// ============================================================================
|
|
|
|
#ifdef USE_CUDA
|
|
#include "backend/cuda/all.cuh"
|
|
#endif
|