From 82fa059a03c8c7d915b5220328adfaf71270f85d Mon Sep 17 00:00:00 2001 From: Hieu Hoang Date: Mon, 14 Jun 2021 19:03:53 -0700 Subject: [PATCH] 'use' variables --- src/layers/logits.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/layers/logits.cpp b/src/layers/logits.cpp index 109f2702..e8fe691b 100644 --- a/src/layers/logits.cpp +++ b/src/layers/logits.cpp @@ -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();