inference-engine/src/inference_engine/builders/ie_concat_layer.cpp

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <builders/ie_concat_layer.hpp>
   6 #include <ie_cnn_layer_builder.h>
   7
   8 #include <vector>
   9 #include <string>
  10
  11 using namespace InferenceEngine;
  12
  13 Builder::ConcatLayer::ConcatLayer(const std::string& name): LayerDecorator("Concat", name) {
  14     getLayer()->getOutputPorts().resize(1);
  15     setAxis(1);
  16 }
  17
  18 Builder::ConcatLayer::ConcatLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
  19     checkType("Concat");
  20 }
  21
  22 Builder::ConcatLayer::ConcatLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
  23     checkType("Concat");
  24 }
  25
  26 Builder::ConcatLayer& Builder::ConcatLayer::setName(const std::string& name) {
  27     getLayer()->setName(name);
  28     return *this;
  29 }
  30
  31 const Port& Builder::ConcatLayer::getOutputPort() const {
  32     return getLayer()->getOutputPorts()[0];
  33 }
  34
  35 Builder::ConcatLayer& Builder::ConcatLayer::setOutputPort(const Port &port) {
  36     getLayer()->getOutputPorts()[0] = port;
  37     return *this;
  38 }
  39
  40 const std::vector<Port>& Builder::ConcatLayer::getInputPorts() const {
  41     return getLayer()->getInputPorts();
  42 }
  43
  44 Builder::ConcatLayer& Builder::ConcatLayer::setInputPorts(const std::vector<Port>& ports) {
  45     getLayer()->getInputPorts() = ports;
  46     return *this;
  47 }
  48
  49 size_t Builder::ConcatLayer::getAxis() const {
  50     return getLayer()->getParameters().at("axis");
  51 }
  52
  53 Builder::ConcatLayer& Builder::ConcatLayer::setAxis(size_t axis) {
  54     getLayer()->getParameters()["axis"] = axis;
  55     return *this;
  56 }
  57
  58 REG_VALIDATOR_FOR(Concat, [] (const InferenceEngine::Builder::Layer::CPtr &input_layer, bool partial) {
  59     if (partial) {
  60         return;
  61     }
  62     Builder::ConcatLayer layer(input_layer);
  63     if (layer.getInputPorts().size() < 1) {
  64         THROW_IE_EXCEPTION << "Layer " << layer.getName() << " contains incorrect input ports. "
  65                            << "It takes at least two Blobs";
  66     }
  67     for (size_t i = 1; i < layer.getInputPorts().size(); ++i) {
  68         if (layer.getInputPorts()[i - 1].shape().size() != layer.getInputPorts()[i].shape().size()) {
  69             THROW_IE_EXCEPTION << "Layer " << layer.getName() << " contains incorrect input ports. "
  70                                << "It should have equal number of dimensions";
  71         }
  72     }
  73     if (layer.getInputPorts()[0].shape().size() != layer.getOutputPort().shape().size()) {
  74         THROW_IE_EXCEPTION << "Layer " << layer.getName() << " contains incorrect input and output ports "
  75                            << "It should have equal number of dimensions";
  76     }
  77     if (layer.getAxis() >= layer.getOutputPort().shape().size()) {
  78         THROW_IE_EXCEPTION << "Layer " << layer.getName() << "contains incorrect axis. "
  79                            << "It should be >= 0 and < number of port's dimensions.";
  80     }
  81     for (size_t i = 0; i < layer.getOutputPort().shape().size(); ++i) {
  82         if (i == layer.getAxis()) {
  83             size_t sumInputDimensions = 0;
  84             for (const Port& port : layer.getInputPorts()) {
  85                 sumInputDimensions += port.shape()[i];
  86             }
  87             if (sumInputDimensions != layer.getOutputPort().shape()[i]) {
  88                 THROW_IE_EXCEPTION << "Layer " << layer.getName() << " contains incorrect input and output ports "
  89                                    << "Sum of input port's dimensions in the given axis should be equal to output ports dimension in the same axis.";
  90             }
  91         } else {
  92             for (const Port& port : layer.getInputPorts()) {
  93                 if (port.shape()[i] != layer.getOutputPort().shape()[i]) {
  94                     THROW_IE_EXCEPTION << "Layer " << layer.getName() << " contains incorrect input and output ports. "
  95                                        << "It should have equal dimensions in axis different from given";
  96                 }
  97             }
  98         }
  99     }
 100 });
 101
 102 REG_CONVERTER_FOR(Concat, [] (const CNNLayerPtr& cnnLayer, Builder::Layer& layer) {
 103     layer.getParameters()["axis"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("axis", 1));
 104 });
 105