ATLAS Offline Software
Loading...
Searching...
No Matches
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

namespace  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)