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

mryzhov · Dec 15, 2020 · e05d435 · e05d435
1 parent d996dbd
commit e05d435
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/...ine/thirdparty/clDNN/kernel_selector/core/actual_kernels/resample/resample_kernel_ref.cpp b/...ine/thirdparty/clDNN/kernel_selector/core/actual_kernels/resample/resample_kernel_ref.cpp
@@ -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;
 }
 

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.cpp b/inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.cpp
@@ -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;
     }

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.h b/inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.h
@@ -180,6 +180,8 @@ class ParamsKey {
                         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;