[IE CLDNN] Fix linear_onnx Interpolate selection (#2769)
authorEgor Churaev <egor.churaev@intel.com>
Fri, 23 Oct 2020 10:16:47 +0000 (13:16 +0300)
committerGitHub <noreply@github.com>
Fri, 23 Oct 2020 10:16:47 +0000 (13:16 +0300)
inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/resample/resample_kernel_ref.cpp
inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.cpp
inference-engine/thirdparty/clDNN/kernel_selector/core/kernel_selector_params.h

index eb66fba..73938d3 100644 (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;
 }
 
index d59da3e..299b784 100644 (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;
     }
index f55490b..9181513 100644 (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;