diff --git a/inference-engine/src/cldnn_engine/cldnn_program.cpp b/inference-engine/src/cldnn_engine/cldnn_program.cpp index a68b40cd7d3..a34ba74f460 100644 --- a/inference-engine/src/cldnn_engine/cldnn_program.cpp +++ b/inference-engine/src/cldnn_engine/cldnn_program.cpp @@ -4981,7 +4981,7 @@ void Program::CreateNonMaxSuppressionPrimitive(cldnn::topology& topology, Infere auto centerPointBox = nonMaxSupression->center_point_box; auto outputIndices = nonMaxSupression->outData[0]->getTensorDesc().getDims()[0]; - auto name = layer_type_name_ID(layer); + auto name = layer->outData.size() > 1 ? (layer_type_lower(layer) + ":" + layer->outData[0]->getName()) : layer_type_name_ID(layer); auto prim = cldnn::non_max_suppression( name, reorderedInputs[0], diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/arg_max_min/arg_max_min_kernel_axis.cpp b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/arg_max_min/arg_max_min_kernel_axis.cpp index 7e2aff58839..257f8250ce5 100644 --- a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/arg_max_min/arg_max_min_kernel_axis.cpp +++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/arg_max_min/arg_max_min_kernel_axis.cpp @@ -62,6 +62,7 @@ ParamsKey ArgMaxMinKernelAxis::GetSupportedKey() const { k.EnableDifferentTypes(); k.EnableBatching(); k.EnableTensorPitches(); + k.EnableTensorOffset(); return k; }