Class Stepwise

Inheritance Relationships

Base Type

Class Documentation

class Stepwise : public marian::IEncoderDecoder

Public Functions

Stepwise(Ptr<IEncoderDecoder> encdec, Ptr<ILogProbStep> cost)
virtual void load(Ptr<ExpressionGraph> graph, const std::vector<io::Item> &items, bool markedReloaded = true)
virtual void load(Ptr<ExpressionGraph> graph, const std::string &name, bool markedReloaded = true)
virtual void mmap(Ptr<ExpressionGraph> graph, const void *ptr, bool markedReloaded = true)
virtual void save(Ptr<ExpressionGraph> graph, const std::string &name, bool saveTranslatorConfig = false)
virtual void clear(Ptr<ExpressionGraph> graph)
virtual Logits build(Ptr<ExpressionGraph> graph, Ptr<data::Batch> batch, bool clearGraph = true)
virtual Ptr<DecoderState> startState(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch)
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)
virtual Logits build(Ptr<ExpressionGraph>, Ptr<data::CorpusBatch>, bool = true)
virtual Ptr<Options> getOptions()
virtual void setShortlistGenerator(Ptr<const data::ShortlistGenerator> shortlistGenerator)
virtual Ptr<data::Shortlist> getShortlist()
virtual data::SoftAlignment getAlignment()

Protected Attributes

Ptr<IEncoderDecoder> encdec_
Ptr<ILogProbStep> cost_