IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GNA] Added fix for export scores (#3009)

Browse Source
This commit is contained in:
Andrey Dmitriev 2020-11-10 16:33:41 +03:00 committed by GitHub
parent dfae8fec65
commit 12bb8abe5c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
inference-engine/samples/speech_sample/main.cpp
View File

@ -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<size_t>());
slog::info << "Number scores per frame : " << numScoresPerFrame << slog::endl;
numFrameElementsInput.resize(numInputArkFiles);
for (size_t i = 0; i < inputArkFiles.size(); i++) {