network.cuh File Reference

#include "cublas_handle.cuh"
#include "device_buffer.cuh"
#include "kernels.cuh"
#include "layer.cuh"
#include "../seed.hpp"
#include <cmath>
#include <random>
#include <utility>
#include <vector>

Include dependency graph for network.cuh:

Classes
class	cuda_mlp::CudaNetwork
	Feed-forward dense network with GPU-backed parameters and gradients. More...

Namespaces
	cuda_mlp