GruDef
#include <popnn/GruDef.hpp>
Definitions for GRU cell operations.
Enums
-
enum BasicGruCellUnit
The units within a basic GRU cell.
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_GRU_CELL_RESET_GATE = 0
-
enumerator BASIC_GRU_CELL_UPDATE_GATE = 1
-
enumerator BASIC_GRU_CELL_CANDIDATE = 2
-
enumerator BASIC_GRU_CELL_NUM_UNITS = 3
-
enumerator BASIC_GRU_CELL_RESET_GATE = 0