From 12bb8abe5cc33b8eba82c5280877775a3d8dc853 Mon Sep 17 00:00:00 2001 From: Andrey Dmitriev Date: Tue, 10 Nov 2020 16:33:41 +0300 Subject: [PATCH] [GNA] Added fix for export scores (#3009) --- inference-engine/samples/speech_sample/main.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/inference-engine/samples/speech_sample/main.cpp b/inference-engine/samples/speech_sample/main.cpp index 44fc1a52546..c9174739b86 100644 --- a/inference-engine/samples/speech_sample/main.cpp +++ b/inference-engine/samples/speech_sample/main.cpp @@ -857,7 +857,10 @@ int main(int argc, char *argv[]) { uint32_t numFramesReference(0), numFrameElementsReference(0), numBytesPerElementReference(0), numBytesReferenceScoreThisUtterance(0); - const uint32_t numScoresPerFrame = ptrOutputBlob.size() / batchSize; + auto dims = outputs.empty() ? cOutputInfo.rbegin()->second->getDims() : cOutputInfo[outputs[next_output]]->getDims(); + const auto numScoresPerFrame = std::accumulate(std::begin(dims), std::end(dims), size_t{1}, std::multiplies()); + + slog::info << "Number scores per frame : " << numScoresPerFrame << slog::endl; numFrameElementsInput.resize(numInputArkFiles); for (size_t i = 0; i < inputArkFiles.size(); i++) {