inference-engine/src/extension/ext_powerfile.cpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include "ext_list.hpp"
   6 #include "ext_base.hpp"
   7
   8 #include <cmath>
   9 #include <string>
  10 #include <vector>
  11
  12 namespace InferenceEngine {
  13 namespace Extensions {
  14 namespace Cpu {
  15
  16 class PowerFileImpl: public ExtLayerBase {
  17 public:
  18     explicit PowerFileImpl(const CNNLayer* layer) {
  19         try {
  20             if (layer->insData.size() != 1 || layer->outData.empty())
  21                 THROW_IE_EXCEPTION << "Incorrect number of input/output edges!";
  22
  23             // TODO: load this from some file or as blob?
  24             shift_.push_back(1);
  25             shift_.push_back(0);
  26             shift_.push_back(0);
  27             shift_.push_back(0);
  28             shift_.push_back(1);
  29             shift_.push_back(0);
  30
  31             addConfig(layer, {DataConfigurator(ConfLayout::PLN)}, {DataConfigurator(ConfLayout::PLN)});
  32         } catch (InferenceEngine::details::InferenceEngineException &ex) {
  33             errorMsg = ex.what();
  34         }
  35     }
  36
  37     StatusCode execute(std::vector<Blob::Ptr>& inputs, std::vector<Blob::Ptr>& outputs,
  38                        ResponseDesc *resp) noexcept override {
  39         if (inputs.size() != 1 || outputs.empty()) {
  40             if (resp) {
  41                 std::string errorMsg = "Incorrect number of input or output edges!";
  42                 errorMsg.copy(resp->msg, sizeof(resp->msg) - 1);
  43             }
  44             return GENERAL_ERROR;
  45         }
  46         float* src_data = inputs[0]->buffer();
  47         float* dst_data = outputs[0]->buffer();
  48
  49         for (size_t i = 0; i < inputs[0]->size(); i++) {
  50             size_t shift_idx = i % shift_.size();
  51             dst_data[i] = src_data[i] + shift_[shift_idx];
  52         }
  53         return OK;
  54     }
  55
  56 private:
  57     std::vector<int> shift_;
  58 };
  59
  60 REG_FACTORY_FOR(ImplFactory<PowerFileImpl>, PowerFile);
  61
  62 }  // namespace Cpu
  63 }  // namespace Extensions
  64 }  // namespace InferenceEngine