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     using parent = typed_primitive_gpu_impl<average_unpooling>;
  30     using parent::parent;
  31
  32 protected:
  33     kernel::kernel_arguments_data get_arguments(typed_primitive_inst<average_unpooling>& instance,
  34                                                         int32_t split) const override {
  35         kernel::kernel_arguments_data args = parent::get_arguments(instance, split);
  36         return args;
  37     }
  38
  39 public:
  40     static primitive_impl* create(const average_unpooling_node& arg) {
  41         auto average_unpooling_params = get_default_params<kernel_selector::average_unpooling_params>(arg);
  42         auto average_unpooling_optional_params =
  43             get_default_optional_params<kernel_selector::average_unpooling_optional_params>(arg.get_program());
  44         auto& params = average_unpooling_params;
  45
  46         auto primitive = arg.get_primitive();
  47         auto stride = primitive->stride;
  48
  49         params.unpoolSize = {
  50             (uint32_t)primitive->size.spatial[0],
  51             (uint32_t)primitive->size.spatial[1],
  52         };
  53
  54         params.unpoolStride = {(uint32_t)stride.spatial[0], (uint32_t)stride.spatial[1]};
  55
  56         auto& kernel_selector = kernel_selector::average_unpooling_kernel_selector::Instance();
  57         auto best_kernels = kernel_selector.GetBestKernels(average_unpooling_params, average_unpooling_optional_params);
  58
  59         CLDNN_ERROR_BOOL(arg.id(),
  60                          "Best_kernel.empty()",
  61                          best_kernels.empty(),
  62                          "Cannot find a proper kernel with this arguments");
  63
  64         auto average_unpool = new average_unpooling_gpu(arg, best_kernels[0]);
  65
  66         return average_unpool;
  67     }
  68 };
  69
  70 namespace detail {
  71
  72 attach_average_unpooling_gpu::attach_average_unpooling_gpu() {
  73     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::yxfb),
  74                                                 average_unpooling_gpu::create);
  75     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::yxfb),
  76                                                 average_unpooling_gpu::create);
  77     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::bfyx),
  78                                                 average_unpooling_gpu::create);
  79     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::bfyx),
  80                                                 average_unpooling_gpu::create);
  81     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::bfyx),
  82                                                 average_unpooling_gpu::create);
  83     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::yxfb),
  84                                                 average_unpooling_gpu::create);
  85     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f32, format::byxf),
  86                                                 average_unpooling_gpu::create);
  87     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::f16, format::byxf),
  88                                                 average_unpooling_gpu::create);
  89     implementation_map<average_unpooling>::add(std::make_tuple(engine_types::ocl, data_types::i8, format::byxf),
  90                                                 average_unpooling_gpu::create);
  91 }
  92
  93 }  // namespace detail
  94 }  // namespace gpu
  95 }  // namespace cldnn