marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-11-30 21:39:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

clean up hypo state interface

Browse Source
This commit is contained in:
Hieu Hoang 2017-02-07 17:12:36 +00:00
parent 6b7aa7577b
commit a4587cb3f2
3 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/plugin/hypo_info.h
View File

@ -23,6 +23,18 @@ struct HypoState
typedef std::vector<HypoState> HypoStates;
////////////////////////////////////////////////////////////////
struct AmunInput : public HypoState
{
AmunInput(const HypoState &hypoState)
:HypoState(hypoState)
{
}
Words phrase;
};
typedef std::vector<AmunInput> AmunInputs;
}

2
src/plugin/nmt.cu
View File

@ -55,7 +55,7 @@ HypoState MosesPlugin::SetSource(const std::vector<size_t>& words) {
return ret;
}
HypoStates MosesPlugin::Score(const HypoStates &inputs)
HypoStates MosesPlugin::Score(const AmunInputs &inputs)
{
HypoStates outputs(inputs.size());

2
src/plugin/nmt.h
View File

@ -34,7 +34,7 @@ class MosesPlugin {
HypoState SetSource(const std::vector<size_t>& words);
HypoStates Score(const HypoStates &inputs);
HypoStates Score(const AmunInputs &inputs);
private:
amunmt::God god_;