ATLAS Offline Software
Classes | Namespaces | Functions
Stack.cxx File Reference
#include "tauRecTools/lwtnn/Stack.h"
#include <Eigen/Dense>
#include <set>
Include dependency graph for Stack.cxx:

Go to the source code of this file.

Classes

struct  lwtDev::LSTMState
 
struct  lwtDev::GRUState
 

Namespaces

 lwtDev
 

Functions

ILayerlwtDev::get_raw_activation_layer (ActivationConfig activation)
 
std::function< double(double)> lwtDev::get_activation (lwtDev::ActivationConfig act)
 
double lwtDev::nn_sigmoid (double x)
 
double lwtDev::nn_hard_sigmoid (double x)
 
double lwtDev::nn_tanh (double x)
 
double lwtDev::nn_relu (double x)
 
MatrixXd lwtDev::build_matrix (const std::vector< double > &weights, size_t n_inputs)
 
VectorXd lwtDev::build_vector (const std::vector< double > &bias)
 
void lwtDev::throw_if_not_maxout (const LayerConfig &layer)
 
void lwtDev::throw_if_not_dense (const LayerConfig &layer)
 
void lwtDev::throw_if_not_normalization (const LayerConfig &layer)
 
DenseComponents lwtDev::get_component (const lwtDev::LayerConfig &layer, size_t n_in)