IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE CLDNN] Minor fixed to NMS and TopK (#3224)

Browse Source
This commit is contained in:
Vladimir Paramuzov
2020-11-20 15:37:32 +03:00
committed by GitHub
parent a05b7c76b2
commit 1e392b0180
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/cldnn_engine/cldnn_program.cpp
View File

@@ -4981,7 +4981,7 @@ void Program::CreateNonMaxSuppressionPrimitive(cldnn::topology& topology, Infere
auto centerPointBox = nonMaxSupression->center_point_box; auto centerPointBox = nonMaxSupression->center_point_box;
auto outputIndices = nonMaxSupression->outData[0]->getTensorDesc().getDims()[0]; 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( auto prim = cldnn::non_max_suppression(
name, name,
reorderedInputs[0], reorderedInputs[0],

1
inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/arg_max_min/arg_max_min_kernel_axis.cpp vendored
View File

@@ -62,6 +62,7 @@ ParamsKey ArgMaxMinKernelAxis::GetSupportedKey() const {
k.EnableDifferentTypes(); k.EnableDifferentTypes();
k.EnableBatching(); k.EnableBatching();
k.EnableTensorPitches(); k.EnableTensorPitches();
k.EnableTensorOffset();
return k; return k;
} }