network_wrapper.hpp File Reference

Backend-agnostic wrapper for CPU/CUDA networks. More...

#include "layer.hpp"
#include "network.hpp"
#include <iostream>
#include <memory>

Include dependency graph for network_wrapper.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	CpuBackend
	Backend tag for CPU implementations. More...
struct	CudaBackend
	Backend tag for CUDA implementations. More...
struct	ActivationToEnum< cpu_mlp::Linear >
	CPU->CUDA activation mapping for Linear. More...
struct	ActivationToEnum< cpu_mlp::Sigmoid >
	CPU->CUDA activation mapping for Sigmoid. More...
struct	ActivationToEnum< cpu_mlp::Tanh >
	CPU->CUDA activation mapping for Tanh. More...
struct	ActivationToEnum< cpu_mlp::ReLU >
	CPU->CUDA activation mapping for ReLU. More...
class	NetworkWrapper< CpuBackend >
	CPU specialization of the network wrapper. More...

Detailed Description

Backend-agnostic wrapper for CPU/CUDA networks.