marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

'use' variables

Browse Source
This commit is contained in:
Hieu Hoang 2021-06-14 19:03:53 -07:00
parent 5362c2cc0e
commit 82fa059a03
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/layers/logits.cpp
View File

@ -122,7 +122,8 @@ Expr Logits::getFactoredLogits(size_t groupIndex,
Expr lastIndices = shortlist->getIndicesExpr();
assert(lastIndices->shape()[0] == currBeamSize || lastIndices->shape()[0] == 1);
assert(lastIndices->shape()[1] == batchSize || lastIndices->shape()[1] == 1);
currBeamSize; currBeamSize;
factorMasks = lambda({lastIndices}, lastIndices->shape(), Type::float32, forward);
const Shape &s = factorMasks->shape();