#include <vector>
#include <type_traits>
#include "bilstm.h"
#include "linear.h"
Go to the source code of this file.
|
| class | deeplima::eigen_impl::Op_BiLSTM_Dense_ArgMax< M, V, T, AuxScalar, IFracBits, WFracBits > |
| | Fusion of several torch modules implemented for inference in Eigen. More...
|
| |
| struct | deeplima::eigen_impl::Op_BiLSTM_Dense_ArgMax< M, V, T, AuxScalar, IFracBits, WFracBits >::workbench_t |
| |
| struct | deeplima::eigen_impl::Op_BiLSTM_Dense_ArgMax< M, V, T, AuxScalar, IFracBits, WFracBits >::params_bilstm_dense_argmax_t |
| |
|
| #define | TANH(x) ( T(2) / ( T(1) + Eigen::exp( T(-2) * (x).array() ) ) - T(1) ) |
| |
| #define | SIGMOID(x) ( T(1) / ( T(1) + Eigen::exp( - (x).array() ) ) ) |
| |
◆ SIGMOID
| #define SIGMOID |
( |
|
x | ) |
( T(1) / ( T(1) + Eigen::exp( - (x).array() ) ) ) |
◆ TANH
| #define TANH |
( |
|
x | ) |
( T(2) / ( T(1) + Eigen::exp( T(-2) * (x).array() ) ) - T(1) ) |