LstmDef.hpp

Go to the documentation of this file.

// Copyright (c) 2017 Graphcore Ltd. All rights reserved.
#ifndef popnn_LstmDef_hpp
#define popnn_LstmDef_hpp
 
enum BasicLstmCellUnit {
  BASIC_LSTM_CELL_FORGET_GATE = 0,
  BASIC_LSTM_CELL_INPUT_GATE = 1,
  BASIC_LSTM_CELL_CANDIDATE = 2,
  BASIC_LSTM_CELL_OUTPUT_GATE = 3,
  BASIC_LSTM_CELL_NUM_UNITS = 4
};
 
#endif // popnn_LstmDef_hpp