Class FastLSTM

Inheritance Relationships

Base Type

Class Documentation

class FastLSTM : public marian::rnn::Cell

Public Functions

FastLSTM(Ptr<ExpressionGraph> graph, Ptr<Options> options)
virtual State apply(std::vector<Expr> inputs, State state, Expr mask = nullptr)
virtual std::vector<Expr> applyInput(std::vector<Expr> inputs)
virtual State applyState(std::vector<Expr> xWs, State state, Expr mask = nullptr)

Protected Attributes

std::string prefix_
Expr U_
Expr W_
Expr b_
Expr gamma1_
Expr gamma2_
bool layerNorm_
float dropout_
Expr dropMaskX_
Expr dropMaskS_
Expr fakeInput_