Class Shortlist

Inheritance Relationships

Derived Type

Class Documentation

class Shortlist

Subclassed by marian::data::LSHShortlist

Public Functions

Shortlist(const std::vector<WordIndex> &indices)
~Shortlist()
virtual bool isDynamic() const
WordIndex reverseMap(int beamIdx, int batchIdx, int idx) const
WordIndex tryForwardMap(WordIndex wIdx) const
void filter(Expr input, Expr weights, bool isLegacyUntransposedW, Expr b, Expr lemmaEt)
Expr getIndicesExpr() const
virtual Expr getCachedShortWt() const
virtual Expr getCachedShortb() const
virtual Expr getCachedShortLemmaEt() const

Public Static Attributes

constexpr WordIndex npos = {std::numeric_limits<WordIndex>::max()}

Protected Functions

void createCachedTensors(Expr weights, bool isLegacyUntransposedW, Expr b, Expr lemmaEt, int k)

Protected Attributes

std::vector<WordIndex> indices_
Expr indicesExpr_
Expr cachedShortWt_
Expr cachedShortb_
Expr cachedShortLemmaEt_
bool initialized_