Publishing 2019 R3 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / fully_connected / fully_connected_block_kernel_base.h

diff --git a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/fully_connected/fully_connected_block_kernel_base.h b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/fully_connected/fully_connected_block_kernel_base.h
index 48f3723..33097aa 100644 (file)
--- a/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/fully_connected/fully_connected_block_kernel_base.h
+++ b/inference-engine/thirdparty/clDNN/kernel_selector/core/actual_kernels/fully_connected/fully_connected_block_kernel_base.h
@@ -15,7 +15,7 @@
 #pragma once
 
 #include "fully_connected_kernel_base.h"
-#include <algorithm>
+
 
 namespace kernel_selector {
 class FullyConnectedBlockKernelBase : public FullyConnectedKernelBase {
@@ -27,15 +27,9 @@ protected:
     JitConstants GetJitConstants(const fully_connected_params& params, const DispatchData& kd) const override;
 
     // how many batches will a single work item compute
-    static size_t GetBatchesPerWorkItem(const fully_connected_params& params) {
-        auto batchSize = params.output.Batch().v;
-        return std::min(batchSize, static_cast<size_t>(32U));
-    }
+    virtual size_t GetBatchesPerWorkItem(const fully_connected_params& params) const;
 
-    static size_t GetLocalGroupsSize(const fully_connected_params& params) {
-        auto batchSize = params.output.Batch().v;
-        return std::max(static_cast<size_t>(1U), batchSize / GetBatchesPerWorkItem(params));
-    }
+    size_t GetLocalGroupsSize(const fully_connected_params& params) const;
 
     // how many neurons for a single batch will a single work item produce
     static size_t GetNeuronsPerWorkItem(const fully_connected_params& params) {
@@ -47,4 +41,4 @@ protected:
             return 1;
     }
 };
-}  // namespace kernel_selector
\ No newline at end of file
+}  // namespace kernel_selector