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

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <builders/ie_output_layer.hpp>
   6
   7 #include <string>
   8
   9 using namespace InferenceEngine;
  10
  11 Builder::OutputLayer::OutputLayer(const std::string& name): LayerDecorator("Output", name) {
  12     getLayer()->getInputPorts().resize(1);
  13 }
  14
  15 Builder::OutputLayer::OutputLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
  16     checkType("Output");
  17 }
  18
  19 Builder::OutputLayer::OutputLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
  20     checkType("Output");
  21 }
  22
  23 Builder::OutputLayer& Builder::OutputLayer::setName(const std::string& name) {
  24     getLayer()->setName(name);
  25     return *this;
  26 }
  27
  28 const Port& Builder::OutputLayer::getPort() const {
  29     return getLayer()->getInputPorts()[0];
  30 }
  31
  32 Builder::OutputLayer& Builder::OutputLayer::setPort(const Port &port) {
  33     getLayer()->getInputPorts()[0] = port;
  34     return *this;
  35 }
  36
  37 REG_VALIDATOR_FOR(Output, [] (const InferenceEngine::Builder::Layer::CPtr& input_layer, bool partial) {});