[ONNX][PyTests] Fix result datatype missmatch (#2596)

This commit is contained in:
Tomasz Socha
2020-11-04 12:19:40 +01:00
committed by GitHub
parent 23188e1b04
commit 7c6794344c
9 changed files with 87 additions and 122 deletions

View File

@@ -120,6 +120,8 @@ void MKLDNNPlugin::MKLDNNInferRequest::InferImpl() {
if (graph->hasMeanImageFor(input.first)) {
// If a mean image exists, we convert the blob and send FP32
copyConvert<float>(InferenceEngine::Precision::FP32, input, convertedInputs);
} else if (dynamic_cast<InferenceEngine::TBlob<bool> *>(input.second.get())) {
pushInput<bool>(input.first, input.second);
} else {
// Instead we can send I8 directly
pushInput<uint8_t>(input.first, input.second);