Class Cell

Inheritance Relationships

Base Type

Derived Types

Class Documentation

class Cell : public marian::rnn::Stackable

Subclassed by marian::rnn::FastLSTM, marian::rnn::GRU, marian::rnn::GRUNematus, marian::rnn::ReLU, marian::rnn::SlowLSTM, marian::rnn::SRU, marian::rnn::SSRU, marian::rnn::StackedCell, marian::rnn::Tanh, marian::rnn::TestLSTM

Public Functions

Cell(Ptr<Options> options)
State apply(std::vector<Expr> inputs, State state, Expr mask = nullptr)
virtual std::vector<Expr> getLazyInputs(Ptr<rnn::RNN> parent)
virtual void marian::rnn::Cell::setLazyInputs(std::vector< std::function< Expr (Ptr< rnn::RNN >)>, lazy)
virtual std::vector<Expr> applyInput(std::vector<Expr> inputs) = 0
virtual State applyState(std::vector<Expr>, State, Expr = nullptr) = 0
virtual void clear()

Protected Attributes

std::vector<std::function<Expr(Ptr<rnn::RNN>)>> lazyInputs_