diff options
author | Patrick Simianer <p@simianer.de> | 2015-01-19 14:39:14 +0100 |
---|---|---|
committer | Patrick Simianer <p@simianer.de> | 2015-01-19 14:39:14 +0100 |
commit | 0673fb55c6af8bc94c9f3e577c514160faf736d3 (patch) | |
tree | ff4edd0e1e816f4d412601689ab510b2d0aaccf7 /decoder/ffset.cc | |
parent | cb9fb7088dde35881516c088db402abe747d49fa (diff) | |
parent | 1bce604809399a0adc581fb0102bff11decf3436 (diff) |
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'decoder/ffset.cc')
-rw-r--r-- | decoder/ffset.cc | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/decoder/ffset.cc b/decoder/ffset.cc index 5820f421..8ba70389 100644 --- a/decoder/ffset.cc +++ b/decoder/ffset.cc @@ -14,6 +14,11 @@ ModelSet::ModelSet(const vector<double>& w, const vector<const FeatureFunction*> for (int i = 0; i < models_.size(); ++i) { model_state_pos_[i] = state_size_; state_size_ += models_[i]->StateSize(); + int num_ignored_bytes = models_[i]->IgnoredStateSize(); + if (num_ignored_bytes > 0) { + ranges_to_erase_.push_back( + {state_size_ - num_ignored_bytes, state_size_}); + } } } @@ -70,3 +75,13 @@ void ModelSet::AddFinalFeatures(const FFState& state, HG::Edge* edge,SentenceMet edge->edge_prob_.logeq(edge->feature_values_.dot(weights_)); } +bool ModelSet::NeedsStateErasure() const { return !ranges_to_erase_.empty(); } + +void ModelSet::EraseIgnoredBytes(FFState* state) const { + // TODO: can we memset? + for (const auto& range : ranges_to_erase_) { + for (int i = range.first; i < range.second; ++i) { + (*state)[i] = 0; + } + } +} |