Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / eltwise / eltwise_kernel_ref.cpp
index 3a77765..6b1e6ea 100644 (file)
@@ -1,5 +1,5 @@
 /*
-// Copyright (c) 2016 Intel Corporation
+// Copyright (c) 2019 Intel Corporation
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -15,7 +15,7 @@
 */
 
 #include "eltwise_kernel_ref.h"
-#include "kernel_selector_utils.h" 
+#include "kernel_selector_utils.h"
 
 namespace kernel_selector {
 
@@ -40,6 +40,8 @@ namespace kernel_selector {
         k.EnableBatching();
         k.EnableInt8Quantization();
         k.EnableOutputCalibration();
+        k.EnableEltwiseStride();
+        k.EnableEltwiseBroadcast();
         return k;
     }
 
@@ -56,7 +58,8 @@ namespace kernel_selector {
             if (params.inputs[i].GetLayout() == DataLayout::fs_bs_yx_bsv4_fsv32)
                 return false;
         }
-        if (params.output.GetLayout() == DataLayout::fs_bs_yx_bsv4_fsv32)
+        if (params.output.GetLayout() == DataLayout::fs_bs_yx_bsv4_fsv32 ||
+            params.output.GetLayout() == DataLayout::b_fs_yx_fsv4)
             return false;
 
         return true;
@@ -66,4 +69,4 @@ namespace kernel_selector {
     {
         return GetCommonKernelsData(params, options);
     }
-}
\ No newline at end of file
+}