Dense< In, Out, Activation > Struct Template Reference

Compile-time dense layer definition. More...

#include <pinn_network.hpp>

Static Public Member Functions
static ENZYME_INLINE void	forward (const Real *p, const Real *in, Real *out)
	Forward pass for a single input vector. More...

Static Public Attributes
static constexpr int	InSize = In
static constexpr int	OutSize = Out
static constexpr int	NumParams = (In * Out) + Out

Detailed Description

template<int In, int Out, typename Activation>
struct Dense< In, Out, Activation >

Compile-time dense layer definition.

Member Function Documentation

forward()

template<int In, int Out, typename Activation >

static ENZYME_INLINE void Dense< In, Out, Activation >::forward ( const Real *  p, const Real *  in, Real *  out  )

inlinestatic

Forward pass for a single input vector.

Member Data Documentation

InSize

template<int In, int Out, typename Activation >

constexpr int Dense< In, Out, Activation >::InSize = In

staticconstexpr

NumParams

template<int In, int Out, typename Activation >

constexpr int Dense< In, Out, Activation >::NumParams = (In * Out) + Out

staticconstexpr

OutSize

template<int In, int Out, typename Activation >

constexpr int Dense< In, Out, Activation >::OutSize = Out

staticconstexpr

---

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

• src/enzyme/pinn_network.hpp