Publishing 2019 R3 content

[platform/upstream/dldt.git] / inference-engine / src / inference_engine / shape_infer / built-in / ie_topk_shape_infer.hpp

diff --git a/inference-engine/src/inference_engine/shape_infer/built-in/ie_topk_shape_infer.hpp b/inference-engine/src/inference_engine/shape_infer/built-in/ie_topk_shape_infer.hpp
index 8846bcb..bf562cf 100644 (file)
--- a/inference-engine/src/inference_engine/shape_infer/built-in/ie_topk_shape_infer.hpp
+++ b/inference-engine/src/inference_engine/shape_infer/built-in/ie_topk_shape_infer.hpp
@@ -54,7 +54,7 @@ public:
             THROW_IE_EXCEPTION << " Incorrect input parameters dimensions and axis number!";
 
         int *src_k = inBlobs[TOPK_K]->cbuffer().as<int *>();
-        if (src_k != nullptr)
+        if (src_k == nullptr)
             THROW_IE_EXCEPTION << " Only const input for 'k' is supported!";
 
         src_k += inBlobs[TOPK_K]->getTensorDesc().getBlockingDesc().getOffsetPadding();