diff --git a/inference-engine/tests/functional/shared_test_classes/src/single_layer/detection_output.cpp b/inference-engine/tests/functional/shared_test_classes/src/single_layer/detection_output.cpp index ac7b89d1c88..1d95dfaa8b6 100644 --- a/inference-engine/tests/functional/shared_test_classes/src/single_layer/detection_output.cpp +++ b/inference-engine/tests/functional/shared_test_classes/src/single_layer/detection_output.cpp @@ -158,7 +158,10 @@ void DetectionOutputLayerTestWithAutoBatching::SetUp() { DetectionOutputAttributes commonAttrs; ParamsWhichSizeDepends specificAttrs; size_t batch; + std::string device; std::tie(commonAttrs, specificAttrs, batch, attrs.objectness_score, targetDevice) = this->GetParam(); + //creating the Auto-batching device with the specific batch size + targetDevice = std::string(CommonTestUtils::DEVICE_BATCH) + ":" + device + "(" + std::to_string(batch) + ")"; std::tie(attrs.num_classes, attrs.background_label_id, attrs.top_k, attrs.keep_top_k, attrs.code_type, attrs.nms_threshold, attrs.confidence_threshold, attrs.clip_after_nms, attrs.clip_before_nms, attrs.decrease_label_id) = commonAttrs;