diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-15 00:23:12 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-15 00:23:12 +0300 |
commit | 31e973836b9349b03da44fb4194e3994a6fb798b (patch) | |
tree | 1d8b45ba84d1f9649e85fafa488fc10445cb002d /src/amun/cpu/nematus/encoder_decoder.cpp | |
parent | 33ebf41448441af0fd5180964ad9519ca7e82b0b (diff) |
uint -> unsigned for gpu
Diffstat (limited to 'src/amun/cpu/nematus/encoder_decoder.cpp')
-rw-r--r-- | src/amun/cpu/nematus/encoder_decoder.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/amun/cpu/nematus/encoder_decoder.cpp b/src/amun/cpu/nematus/encoder_decoder.cpp index 0fdb202f..8edc71bb 100644 --- a/src/amun/cpu/nematus/encoder_decoder.cpp +++ b/src/amun/cpu/nematus/encoder_decoder.cpp @@ -29,7 +29,7 @@ EncoderDecoder::EncoderDecoder(const God &god, {} -void EncoderDecoder::Decode(const State& in, State& out, const std::vector<uint>&) { +void EncoderDecoder::Decode(const State& in, State& out, const std::vector<unsigned>&) { const EDState& edIn = in.get<EDState>(); EDState& edOut = out.get<EDState>(); @@ -54,8 +54,8 @@ void EncoderDecoder::Encode(const Sentences& sources) { void EncoderDecoder::AssembleBeamState(const State& in, const Beam& beam, State& out) { - std::vector<uint> beamWords; - std::vector<uint> beamStateIds; + std::vector<unsigned> beamWords; + std::vector<unsigned> beamStateIds; for(auto h : beam) { beamWords.push_back(h->GetWord()); beamStateIds.push_back(h->GetPrevStateIndex()); @@ -84,7 +84,7 @@ unsigned EncoderDecoder::GetVocabSize() const { } -void EncoderDecoder::Filter(const std::vector<uint>& filterIds) { +void EncoderDecoder::Filter(const std::vector<unsigned>& filterIds) { decoder_->Filter(filterIds); } |