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

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <builders/ie_fully_connected_layer.hpp>
   6 #include <ie_cnn_layer_builder.h>
   7
   8 #include <vector>
   9 #include <string>
  10
  11 using namespace InferenceEngine;
  12
  13 Builder::FullyConnectedLayer::FullyConnectedLayer(const std::string& name): LayerDecorator("FullyConnected", name) {
  14     getLayer()->getInputPorts().resize(3);
  15     getLayer()->getInputPorts()[1].setParameter("type", "weights");
  16     getLayer()->getInputPorts()[2].setParameter("type", "biases");
  17     getLayer()->getOutputPorts().resize(1);
  18     getLayer()->getParameters()["out-size"] = 0;
  19 }
  20
  21 Builder::FullyConnectedLayer::FullyConnectedLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
  22     checkType("FullyConnected");
  23 }
  24
  25 Builder::FullyConnectedLayer::FullyConnectedLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
  26     checkType("FullyConnected");
  27 }
  28
  29 Builder::FullyConnectedLayer &Builder::FullyConnectedLayer::setName(const std::string &name) {
  30     getLayer()->setName(name);
  31     return *this;
  32 }
  33
  34 const Port& Builder::FullyConnectedLayer::getInputPort() const {
  35     return getLayer()->getInputPorts()[0];
  36 }
  37
  38 Builder::FullyConnectedLayer& Builder::FullyConnectedLayer::setInputPort(const Port& port) {
  39     getLayer()->getInputPorts()[0] = port;
  40     return *this;
  41 }
  42
  43 const Port& Builder::FullyConnectedLayer::getOutputPort() const {
  44     return getLayer()->getOutputPorts()[0];
  45 }
  46
  47 Builder::FullyConnectedLayer& Builder::FullyConnectedLayer::setOutputPort(const Port& port) {
  48     getLayer()->getOutputPorts()[0] = port;
  49     return *this;
  50 }
  51
  52 size_t Builder::FullyConnectedLayer::getOutputNum() const {
  53     return getLayer()->getParameters().at("out-size");
  54 }
  55
  56 Builder::FullyConnectedLayer& Builder::FullyConnectedLayer::setOutputNum(size_t outNum) {
  57     getLayer()->getParameters()["out-size"] = outNum;
  58     return *this;
  59 }
  60
  61 REG_VALIDATOR_FOR(FullyConnected, [](const InferenceEngine::Builder::Layer::CPtr& layer, bool partial) {
  62 });
  63
  64 REG_CONVERTER_FOR(FullyConnected, [](const CNNLayerPtr& cnnLayer, Builder::Layer& layer) {
  65     layer.getParameters()["out-size"] = static_cast<size_t>(cnnLayer->GetParamAsUInt("out-size", 0));
  66 });