Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / src / extension / ext_argmax.cpp

diff --git a/inference-engine/src/extension/ext_argmax.cpp b/inference-engine/src/extension/ext_argmax.cpp
index c6efa6c..3a8dab3 100644 (file)
--- a/inference-engine/src/extension/ext_argmax.cpp
+++ b/inference-engine/src/extension/ext_argmax.cpp
@@ -1,4 +1,4 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
@@ -23,7 +23,7 @@ public:
             if (layer->insData.size() != 1 || layer->outData.empty())
                 THROW_IE_EXCEPTION << "Incorrect number of input/output edges!";
 
-            out_max_val_ = static_cast<bool>(layer->GetParamAsInt("out_max_val"));
+            out_max_val_ = layer->GetParamAsBool("out_max_val", false);
             top_k_       = layer->GetParamAsInt("top_k");
 
             has_axis_ = (layer->params.find("axis") != layer->params.end());
@@ -73,12 +73,12 @@ public:
                         dst_data[(i / axis_dist * top_k_ + j) * axis_dist + i % axis_dist] = src_vector[j].first;
                     } else {
                         // Produces max_ind and max_val
-                        dst_data[2 * i * top_k_ + j] = src_vector[j].second;
+                        dst_data[2 * i * top_k_ + j] = static_cast<float>(src_vector[j].second);
                         dst_data[2 * i * top_k_ + top_k_ + j] = src_vector[j].first;
                     }
                 } else {
                     // Produces max_ind per axis
-                    dst_data[(i / axis_dist * top_k_ + j) * axis_dist + i % axis_dist] = src_vector[j].second;
+                    dst_data[(i / axis_dist * top_k_ + j) * axis_dist + i % axis_dist] = static_cast<float>(src_vector[j].second);
                 }
             }
         }