NetworkWrapper< CpuBackend > Class Reference

CPU specialization of the network wrapper. More...

#include <network_wrapper.hpp>

Public Types
using	InternalNetwork = cpu_mlp::Network

Public Member Functions
	NetworkWrapper ()=default
template<int In, int Out, typename Activation >
void	addLayer ()
void	bindParams ()
void	bindParams (unsigned int seed)
InternalNetwork &	getInternal ()
	Access the underlying CPU network. More...
const InternalNetwork &	getInternal () const
	Access the underlying CPU network (const). More...
size_t	getParamsSize () const
	Total number of parameters. More...

Detailed Description

CPU specialization of the network wrapper.

Member Typedef Documentation

InternalNetwork

using NetworkWrapper< CpuBackend >::InternalNetwork = cpu_mlp::Network

Constructor & Destructor Documentation

NetworkWrapper()

NetworkWrapper< CpuBackend >::NetworkWrapper ( )

default

Member Function Documentation

addLayer()

template<int In, int Out, typename Activation >

void NetworkWrapper< CpuBackend >::addLayer ( )

inline

bindParams() [1/2]

void NetworkWrapper< CpuBackend >::bindParams ( )

inline

bindParams() [2/2]

void NetworkWrapper< CpuBackend >::bindParams ( unsigned int  seed )

inline

getInternal() [1/2]

InternalNetwork& NetworkWrapper< CpuBackend >::getInternal ( )

inline

Access the underlying CPU network.

Here is the caller graph for this function:

getInternal() [2/2]

const InternalNetwork& NetworkWrapper< CpuBackend >::getInternal ( ) const

inline

Access the underlying CPU network (const).

getParamsSize()

size_t NetworkWrapper< CpuBackend >::getParamsSize ( ) const

inline

Total number of parameters.

---

The documentation for this class was generated from the following file:

• src/network_wrapper.hpp