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

[IE CLDNN] Fix linear_onnx Interpolate selection (#2769)

Browse Source
This commit is contained in:
Egor Churaev 2020-10-23 13:16:47 +03:00 committed by GitHub
parent 1bae540895
commit ca95240c91
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/resample/resample_kernel_ref.cpp vendored
View File

@ -41,6 +41,7 @@ ParamsKey ResampleKernelRef::GetSupportedKey() const {
k.EnableReampleType(ResampleType::CAFFE_BILINEAR_INTERP);
k.EnableReampleType(ResampleType::BILINEAR_INTERP);
k.EnableReampleType(ResampleType::CUBIC);
k.EnableReampleType(ResampleType::LINEAR_ONNX);
return k;
}

8
inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.cpp vendored
View File

@ -1,4 +1,4 @@
// Copyright (c) 2016-2019 Intel Corporation
// Copyright (c) 2016-2020 Intel Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
@ -325,6 +325,12 @@ void ParamsKey::EnableReampleType(ResampleType a) {
case ResampleType::BILINEAR_INTERP:
key.restrict.val.dedicated.resample.bilinear_interp = 1;
break;
case ResampleType::CUBIC:
key.restrict.val.dedicated.resample.cubic = 1;
break;
case ResampleType::LINEAR_ONNX:
key.restrict.val.dedicated.resample.linear_onnx = 1;
break;
default:
break;
}

2
inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.h vendored
View File

@ -180,6 +180,8 @@ public:
uint32_t nearest_neighbor : 1;
uint32_t caffe_bilinear_interp : 1;
uint32_t bilinear_interp : 1;
uint32_t cubic : 1;
uint32_t linear_onnx : 1;
} resample;
struct reorder_t {
uint32_t winograd : 1;