Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / kernel_selector / core / actual_kernels / roi_pooling / roi_pooling_kernel_selector.cpp
index 9dbb71c..969362d 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.
 
 #include "roi_pooling_kernel_selector.h"
 #include "roi_pooling_kernel_ref.h"
-namespace kernel_selector 
+#include "roi_pooling_kernel_ps_ref.h"
+
+namespace kernel_selector
 {
     roi_pooling_kernel_selector::roi_pooling_kernel_selector()
     {
         Attach<ROIPoolingKernelRef>();
+        Attach<PSROIPoolingKernelRef>();
     }
 
     KernelsData roi_pooling_kernel_selector::GetBestKernels(const Params& params, const optional_params& options) const
     {
         return GetNaiveBestKernel(params, options, KernelType::ROI_POOLING);
     }
-}
\ No newline at end of file
+}