Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / roi_pooling / roi_pooling_kernel_ref.h
index 1bc3c60..e4c8934 100644 (file)
@@ -1,5 +1,5 @@
 /*
-// Copyright (c) 2016 Intel Corporation
+// Copyright (c) 2016-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.
 
 #pragma once
 
-#include "common_kernel_base.h"
+#include "roi_pooling_kernel_base.h"
 
 namespace kernel_selector
 {
-    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-    // roi_pooling_params
-    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-    struct roi_pooling_params : public base_params
-    {
-        roi_pooling_params() : base_params(KernelType::ROI_POOLING) {}
-
-        PoolType    mode = PoolType::MAX;
-        size_t      pooledWidth = 0;
-        size_t      pooledHeight = 0;
-        size_t      groupSize = 0;
-        float       spatialScale = 1.f;
-
-        virtual ParamsKey GetParamsKey() const
-        {
-            return base_params::GetParamsKey();
-        }
-    };
-
-    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-    // roi_pooling_optional_params
-    ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-    struct roi_pooling_optional_params : optional_params
-    {
-        roi_pooling_optional_params() : optional_params(KernelType::ROI_POOLING) {}
-    };
 
     ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
     // ROIPoolingKernelRef
     ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
-    class ROIPoolingKernelRef : public common_kernel_base
+    class ROIPoolingKernelRef : public ROIPoolingKernelBase
     {
     public:
-        ROIPoolingKernelRef() : common_kernel_base("roi_pooling_ref") {}
+        ROIPoolingKernelRef() : ROIPoolingKernelBase("roi_pooling_ref") {}
         virtual ~ROIPoolingKernelRef() {}
 
         using DispatchData = CommonDispatchData;
 
         virtual KernelsData GetKernelsData(const Params& params, const optional_params& options) const override;
-        virtual ParamsKey GetSupportedKey() const override;
 
     protected:
-        JitConstants GetJitConstants(const roi_pooling_params& params) const;
+        virtual ParamsKey GetSupportedKey() const override;
     };
-}
\ No newline at end of file
+}