Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / src / arg_max_min.cpp

diff --git a/inference-engine/thirdparty/clDNN/src/arg_max_min.cpp b/inference-engine/thirdparty/clDNN/src/arg_max_min.cpp
index aa2f0e4..96cdca3 100644 (file)
--- a/inference-engine/thirdparty/clDNN/src/arg_max_min.cpp
+++ b/inference-engine/thirdparty/clDNN/src/arg_max_min.cpp
@@ -31,7 +31,10 @@ namespace cldnn
 
        layout arg_max_min_inst::calc_output_layout(arg_max_min_node const& node)
        {
-               auto desc = node.get_primitive();
+        assert((bool)node.get_primitive()->output_data_type == false
+               && "Output data type forcing is not supported for "
+                  "arg_max_min_node!");
+        auto desc = node.get_primitive();
 
                auto input_layout = node.input().get_output_layout();