[ci-skip][IE TESTS] removing bathFlag

openvinotoolkit · Jun 29, 2020 · 7a8d9a0 · 7a8d9a0
1 parent bdb8f4a
commit 7a8d9a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 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
@@ -41,7 +41,7 @@ void LayerTestsCommon::Compare(const std::vector<std::uint8_t> &expected, const
     const auto &precision = actual->getTensorDesc().getPrecision();
     auto resize = 1;
     // With dynamic batch, you need to size
-    if (DynamicBathFlag) {
+    if (configuration.count(InferenceEngine::PluginConfigParams::KEY_DYN_BATCH_ENABLED)) {
         resize = actual->getTensorDesc().getDims()[0];
     }
     const auto &size = (actual->size() * bathSize / resize);
@@ -62,9 +62,6 @@ void LayerTestsCommon::Compare(const std::vector<std::uint8_t> &expected, const
 void LayerTestsCommon::ConfigurePlugin() {
     if (!configuration.empty()) {
         core->SetConfig(configuration, targetDevice);
-        if (configuration.count(InferenceEngine::PluginConfigParams::KEY_DYN_BATCH_ENABLED)) {
-            DynamicBathFlag = true;
-        }
     }
 }
 
@@ -104,7 +101,7 @@ void LayerTestsCommon::Infer() {
         inferRequest.SetBlob(info->name(), blob);
         inputs.push_back(blob);
     }
-    if (DynamicBathFlag) {
+    if (configuration.count(InferenceEngine::PluginConfigParams::KEY_DYN_BATCH_ENABLED)) {
         bathSize = cnnNetwork.getInputsInfo().begin()->second->getTensorDesc().getDims()[0] / 2;
         inferRequest.SetBatch(bathSize);
     }

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;
-    bool DynamicBathFlag = false;  // check for dynamic batch
     int bathSize = 1;  // dynamic bath size ( first dimension in inputShapes > 1 !!!)
     virtual void Validate();