Class Shortlist¶

Defined in File shortlist.h

Inheritance Relationships¶

Derived Type¶

public marian::data::LSHShortlist (Class LSHShortlist)

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_¶