inference-engine/thirdparty/clDNN/src/gpu/average_unpooling_gpu.cpp

   1 /*
   2 // Copyright (c) 2018 Intel Corporation
   3 //
   4 // Licensed under the Apache License, Version 2.0 (the "License");
   5 // you may not use this file except in compliance with the License.
   6 // You may obtain a copy of the License at
   7 //
   8 //      http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 // Unless required by applicable law or agreed to in writing, software
  11 // distributed under the License is distributed on an "AS IS" BASIS,
  12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 // See the License for the specific language governing permissions and
  14 // limitations under the License.
  15 */
  16
  17 #include "average_unpooling_inst.h"
  18 #include "primitive_gpu_base.h"
  19 #include "implementation_map.h"
  20 #include "error_handler.h"
  21 #include "kernel_selector_helper.h"
  22 #include "average_unpooling/average_unpooling_kernel_selector.h"
  23 #include "average_unpooling/average_unpooling_kernel_base.h"
  24
  25 namespace cldnn {
  26     namespace gpu {
  27
  28         struct average_unpooling_gpu : typed_primitive_gpu_impl<average_unpooling>
  29         {
  30             using parent = typed_primitive_gpu_impl<average_unpooling>;
  31             using parent::parent;
  32
  33         protected:
  34
  35             virtual kernel::kernel_arguments_data get_arguments(typed_primitive_inst<average_unpooling>& instance, int32_t split) const override
  36             {
  37                 kernel::kernel_arguments_data args = parent::get_arguments(instance, split);
  38                 return args;
  39             }
  40
  41         public:
  42
  43             static primitive_impl* create(const average_unpooling_node& arg)
  44             {
  45                 auto average_unpooling_params = get_default_params<kernel_selector::average_unpooling_params>(arg);
  46                 auto average_unpooling_optional_params = get_default_optional_params<kernel_selector::average_unpooling_optional_params>(arg.get_program());
  47                 auto& params = average_unpooling_params;
  48
  49                 auto primitive = arg.get_primitive();
  50                 auto stride = primitive->stride;
  51
  52                 params.unpoolSize = {
  53                     (uint32_t)primitive->size.spatial[0],
  54                     (uint32_t)primitive->size.spatial[1],
  55                 };
  56
  57                 params.unpoolStride = {
  58                     (uint32_t)stride.spatial[0],
  59                     (uint32_t)stride.spatial[1]
  60                 };
  61
  62                 auto& kernel_selector = kernel_selector::average_unpooling_kernel_selector::Instance();
  63                 auto best_kernels = kernel_selector.GetBestKernels(average_unpooling_params, average_unpooling_optional_params);
  64
  65                 CLDNN_ERROR_BOOL(arg.id(), "Best_kernel.empty()", best_kernels.empty(), "Cannot find a proper kernel with this arguments");
  66
  67                 auto average_unpool = new average_unpooling_gpu(arg, best_kernels[0]);
  68
  69                 return average_unpool;
  70             }
  71         };
  72
  73         namespace {
  74             struct attach {
  75                 attach()
  76                 {
  77                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::yxfb), average_unpooling_gpu::create);
  78                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::yxfb), average_unpooling_gpu::create);
  79                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::bfyx), average_unpooling_gpu::create);
  80                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::bfyx), average_unpooling_gpu::create);
  81                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::bfyx), average_unpooling_gpu::create);
  82                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::yxfb), average_unpooling_gpu::create);
  83                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::byxf), average_unpooling_gpu::create);
  84                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::byxf), average_unpooling_gpu::create);
  85                     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::byxf), average_unpooling_gpu::create);
  86                 }
  87                 ~attach() {}
  88             };
  89             attach attach_impl;
  90         }
  91     }
  92 }