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

   1 // Copyright (C) 2018-2019 Intel Corporation
   2 // SPDX-License-Identifier: Apache-2.0
   3 //
   4
   5 #include <builders/ie_const_layer.hpp>
   6
   7 #include <string>
   8
   9 using namespace InferenceEngine;
  10
  11 Builder::ConstLayer::ConstLayer(const std::string& name): LayerDecorator("Const", name) {
  12     getLayer()->getOutputPorts().resize(1);
  13     getLayer()->getParameters()["custom"] = Blob::CPtr();
  14 }
  15
  16 Builder::ConstLayer::ConstLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
  17     checkType("Const");
  18 }
  19
  20 Builder::ConstLayer::ConstLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
  21     checkType("Const");
  22 }
  23
  24 Builder::ConstLayer& Builder::ConstLayer::setName(const std::string& name) {
  25     getLayer()->setName(name);
  26     return *this;
  27 }
  28
  29 const Port& Builder::ConstLayer::getPort() const {
  30     return getLayer()->getOutputPorts()[0];
  31 }
  32
  33 Builder::ConstLayer& Builder::ConstLayer::setPort(const Port &port) {
  34     const auto & data = getLayer()->getOutputPorts()[0].getData();
  35     getLayer()->getOutputPorts()[0] = port;
  36     getLayer()->getOutputPorts()[0].setData(data);
  37     return *this;
  38 }
  39
  40 Builder::ConstLayer& Builder::ConstLayer::setData(const Blob::CPtr& data) {
  41     getLayer()->getParameters()["custom"] = data;
  42     getLayer()->getOutputPorts()[0].getData()->setData(std::const_pointer_cast<Blob>(data));
  43     return *this;
  44 }
  45
  46 const Blob::CPtr& Builder::ConstLayer::getData() const {
  47     if (getLayer()->getParameters().at("custom").as<Blob::CPtr>().get() !=
  48             getLayer()->getOutputPorts()[0].getData()->getData().get())
  49         THROW_IE_EXCEPTION << "Constant data output port has incorrect data!";
  50     return getLayer()->getParameters().at("custom").as<Blob::CPtr>();
  51 }
  52
  53 REG_VALIDATOR_FOR(Const, [] (const InferenceEngine::Builder::Layer::CPtr& layer, bool partial)  {
  54     Builder::ConstLayer constBuilder(layer);
  55     const auto& data = constBuilder.getData();
  56     if (!data || data->cbuffer() == nullptr)
  57         THROW_IE_EXCEPTION << "Cannot create Const layer! Data is required!";
  58 });