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

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

Browse Source
This commit is contained in:
Vladimir Paramuzov 2020-11-20 15:37:55 +03:00 committed by GitHub
parent f3358ed89d
commit 27be33ba53
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show Stats 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 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],

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.EnableBatching();
k.EnableTensorPitches();
k.EnableTensorOffset();
return k;
}