Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/models/states.h')
-rw-r--r--src/models/states.h6
1 files changed, 4 insertions, 2 deletions
diff --git a/src/models/states.h b/src/models/states.h
index f1f3c704..e2635c87 100644
--- a/src/models/states.h
+++ b/src/models/states.h
@@ -48,8 +48,10 @@ public:
virtual Expr getProbs() { return probs_; }
virtual void setProbs(Expr probs) { probs_ = probs; }
- virtual Ptr<DecoderState> select(const std::vector<size_t>& selIdx, int beamSize) {
- return New<DecoderState>(states_.select(selIdx, beamSize), probs_, encStates_);
+ virtual Ptr<DecoderState> select(const std::vector<size_t>& selIdx,
+ int beamSize) {
+ return New<DecoderState>(
+ states_.select(selIdx, beamSize), probs_, encStates_);
}
virtual const rnn::States& getStates() { return states_; }