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:
authorHieu Hoang <hieuhoang@gmail.com>2017-11-15 19:26:01 +0300
committerHieu Hoang <hieuhoang@gmail.com>2017-11-15 19:26:01 +0300
commit8c89af1edf01e241074eb84d03675d6b5725f7c9 (patch)
tree7f84d28957986779590900426507d3520a7437b5 /src/amun/cpu/decoder/encoder_decoder.h
parent981325e7d7eaee47700521622c7a838b293b8430 (diff)
add GetB4() and GetNBest()
Diffstat (limited to 'src/amun/cpu/decoder/encoder_decoder.h')
-rw-r--r--src/amun/cpu/decoder/encoder_decoder.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/src/amun/cpu/decoder/encoder_decoder.h b/src/amun/cpu/decoder/encoder_decoder.h
index c8bb56b4..a24a6ee0 100644
--- a/src/amun/cpu/decoder/encoder_decoder.h
+++ b/src/amun/cpu/decoder/encoder_decoder.h
@@ -25,6 +25,18 @@ class CPUEncoderDecoderBase : public Scorer {
virtual void GetAttention(mblas::Matrix& Attention) = 0;
virtual mblas::Matrix& GetAttention() = 0;
+ virtual void *GetNBest()
+ {
+ assert(false);
+ return nullptr;
+ }
+
+ virtual const BaseMatrix *GetB4() const
+ {
+ assert(false);
+ return nullptr;
+ }
+
protected:
mblas::Matrix SourceContext_;
};