[IE TESTS] removing bathSize

openvinotoolkit · Jul 3, 2020 · f6ed690 · f6ed690
1 parent 0b5e003
commit f6ed690
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/inference-engine/tests/ie_test_utils/functional_test_utils/layer_test_utils.cpp b/inference-engine/tests/ie_test_utils/functional_test_utils/layer_test_utils.cpp
@@ -40,9 +40,11 @@ void LayerTestsCommon::Compare(const std::vector<std::uint8_t> &expected, const
 
     const auto &precision = actual->getTensorDesc().getPrecision();
     auto resize = 1;
+    auto bathSize = 1;
     // With dynamic batch, you need to size
     if (configuration.count(InferenceEngine::PluginConfigParams::KEY_DYN_BATCH_ENABLED)) {
         resize = actual->getTensorDesc().getDims()[0];
+        bathSize = resize > 1 ? resize/ 2 : 1;
     }
     const auto &size = (actual->size() * bathSize / resize);
     switch (precision) {
@@ -102,7 +104,7 @@ void LayerTestsCommon::Infer() {
         inputs.push_back(blob);
     }
     if (configuration.count(InferenceEngine::PluginConfigParams::KEY_DYN_BATCH_ENABLED)) {
-        bathSize = cnnNetwork.getInputsInfo().begin()->second->getTensorDesc().getDims()[0] / 2;
+        auto bathSize = cnnNetwork.getInputsInfo().begin()->second->getTensorDesc().getDims()[0] / 2;
         inferRequest.SetBatch(bathSize);
     }
     inferRequest.Infer();

diff --git a/inference-engine/tests/ie_test_utils/functional_test_utils/layer_test_utils.hpp b/inference-engine/tests/ie_test_utils/functional_test_utils/layer_test_utils.hpp
@@ -102,7 +102,6 @@ class LayerTestsCommon : public CommonTestUtils::TestsCommon {
     std::vector<InferenceEngine::Blob::Ptr> inputs;
     float threshold;
     InferenceEngine::CNNNetwork cnnNetwork;
-    int bathSize = 1;  // dynamic bath size ( first dimension in inputShapes > 1 !!!)
     virtual void Validate();
 
     virtual std::vector<std::vector<std::uint8_t>> CalculateRefs();