Program Listing for File encoder.h

↰ Return to documentation for file (src/models/encoder.h)

#pragma once

#include "marian.h"
#include "models/states.h"

namespace marian {

class EncoderBase : public EncoderDecoderLayerBase {
public:
  EncoderBase(Ptr<ExpressionGraph> graph, Ptr<Options> options) :
    EncoderDecoderLayerBase(graph, options, "encoder", /*batchIndex=*/0,
        options->get<float>("dropout-src", 0.0f),
        options->get<bool>("embedding-fix-src", false)) {}

  // @TODO: turn into an interface. Also see if we can get rid of the graph parameter.
  virtual Ptr<EncoderState> build(Ptr<ExpressionGraph>, Ptr<data::CorpusBatch>) = 0;

  virtual void clear() = 0;
};

}  // namespace marian