Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / mkl-dnn / tests / gtests / test_batch_normalization.cpp

diff --git a/inference-engine/thirdparty/mkl-dnn/tests/gtests/test_batch_normalization.cpp b/inference-engine/thirdparty/mkl-dnn/tests/gtests/test_batch_normalization.cpp
index 48d5bfc..8b82ddb 100644 (file)
--- a/inference-engine/thirdparty/mkl-dnn/tests/gtests/test_batch_normalization.cpp
+++ b/inference-engine/thirdparty/mkl-dnn/tests/gtests/test_batch_normalization.cpp
@@ -152,7 +152,7 @@ void check_bnrm_bwd(const test_bnrm_params_t &p,
 {
     const test_bnrm_sizes_t &bp = p.sizes;
     const bool use_weights = flags & use_scale_shift;
-    const bool calculate_diff_stats = !(flags & omit_stats);
+    const bool calculate_diff_stats = !(flags & use_global_stats);
 
     const data_t *src_data = (const data_t *)src.get_data_handle();
     const data_t *weights_data = use_weights ? (const data_t *)weights.get_data_handle() : nullptr;
@@ -316,11 +316,11 @@ protected:
         Forward(use_scale_shift | use_global_stats, training);
 
         Backward(0u, backward_data);
-        Backward(omit_stats, backward_data);
+        Backward(use_global_stats, backward_data);
         Backward(use_scale_shift, backward);
         Backward(use_scale_shift, backward_data);
-        Backward(use_scale_shift | omit_stats, backward);
-        Backward(use_scale_shift | omit_stats, backward_data);
+        Backward(use_scale_shift | use_global_stats, backward);
+        Backward(use_scale_shift | use_global_stats, backward_data);
 
     }