LstmDef

#include <popnn/LstmDef.hpp>

Definitions for LSTM cell operations.

Enums

enum BasicLstmCellUnit

The units within a basic LSTM cell.

The term unit is used to refer to either a gate, or a cell state vector computation. In general all of these require a weight matrix, a bias and a non-linearity. Typically, a fixed type of non-linearity is associated with each type of unit.

Values:

enumerator BASIC_LSTM_CELL_FORGET_GATE = 0

enumerator BASIC_LSTM_CELL_INPUT_GATE = 1

enumerator BASIC_LSTM_CELL_CANDIDATE = 2

enumerator BASIC_LSTM_CELL_OUTPUT_GATE = 3

enumerator BASIC_LSTM_CELL_NUM_UNITS = 4