Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / convolution / convolution_kernel_bfyx_depthwise_weights_lwg.cpp
index f6841db..12478e8 100644 (file)
@@ -15,8 +15,7 @@
 */
 
 #include "convolution_kernel_bfyx_depthwise_weights_lwg.h"
-#include "kernel_selector_utils.h"
+
 namespace kernel_selector 
 {
     ParamsKey ConvolutionKernel_bfyx_depthwise_weights_lwg::GetSupportedKey() const
@@ -39,6 +38,7 @@ namespace kernel_selector
         k.EnableSubGroup();
         k.EnableSubGroupShort();
         k.EnableDepthwiseSeparableOpt();
+        k.EnableDilation();
         return k;
     }
 
@@ -51,12 +51,11 @@ namespace kernel_selector
         }
 
        const convolution_params& cp = static_cast<const convolution_params&>(p);
-       if (!cp.depthwiseSeparableOpt)
+       if (!cp.depthwise_separable_opt)
            return false;
-
        if ((cp.filterSize.x > 4) ||
            (cp.filterSize.y > 4) ||
-           (cp.inputs[0].Feature().v != cp.split))
+           ((cp.inputs[0].Feature().v != cp.split) && (cp.inputs[0].Feature().v != cp.groups)))
        {
            return false;
        }
@@ -95,6 +94,6 @@ namespace kernel_selector
 
     KernelsData ConvolutionKernel_bfyx_depthwise_weights_lwg::GetKernelsData(const Params& params, const optional_params& options) const
     {
-        return GetCommonKernelsData(params, options);
+        return GetTunedKernelsDataByIndex(params, options);
     }
 }
\ No newline at end of file