Class IEncoderDecoder¶

Defined in File encoder_decoder.h

Inheritance Relationships¶

Base Type¶

public IModel

Derived Types¶

public marian::EncoderDecoder (Class EncoderDecoder)
public marian::models::Stepwise (Class Stepwise)

Class Documentation¶

class IEncoderDecoder : public IModel¶

Subclassed by marian::EncoderDecoder, marian::models::Stepwise

Public Functions

virtual ~IEncoderDecoder()¶

virtual void load(Ptr<ExpressionGraph> graph, const std::vector<io::Item> &items, bool markedReloaded = true) = 0¶

virtual void load(Ptr<ExpressionGraph> graph, const std::string &name, bool markedReloaded = true) = 0¶

virtual void mmap(Ptr<ExpressionGraph> graph, const void *ptr, bool markedReloaded = true) = 0¶

virtual void save(Ptr<ExpressionGraph> graph, const std::string &name, bool saveTranslatorConfig = false) = 0¶

virtual void clear(Ptr<ExpressionGraph> graph) = 0¶

virtual Logits build(Ptr<ExpressionGraph> graph, Ptr<data::Batch> batch, bool clearGraph = true) = 0¶

virtual Logits build(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, bool clearGraph = true) = 0¶

virtual Ptr<DecoderState> startState(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch) = 0¶

virtual Ptr<DecoderState> step(Ptr<ExpressionGraph> graph, Ptr<DecoderState> state, const std::vector<IndexType> &hypIndices, const Words &words, const std::vector<IndexType> &batchIndices, int beamSize) = 0¶

virtual Ptr<Options> getOptions() = 0¶

virtual void setShortlistGenerator(Ptr<const data::ShortlistGenerator> shortlistGenerator) = 0¶

virtual Ptr<data::Shortlist> getShortlist() = 0¶

virtual data::SoftAlignment getAlignment() = 0¶