1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
5 #include <builders/ie_scale_shift_layer.hpp>
9 using namespace InferenceEngine;
11 Builder::ScaleShiftLayer::ScaleShiftLayer(const std::string& name): LayerDecorator("ScaleShift", name) {
12 getLayer()->getInputPorts().resize(3);
13 getLayer()->getInputPorts()[1].setParameter("type", "weights");
14 getLayer()->getInputPorts()[2].setParameter("type", "biases");
15 getLayer()->getOutputPorts().resize(1);
18 Builder::ScaleShiftLayer::ScaleShiftLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
19 checkType("ScaleShift");
22 Builder::ScaleShiftLayer::ScaleShiftLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
23 checkType("ScaleShift");
26 Builder::ScaleShiftLayer& Builder::ScaleShiftLayer::setName(const std::string& name) {
27 getLayer()->setName(name);
31 const Port& Builder::ScaleShiftLayer::getPort() const {
32 return getLayer()->getOutputPorts()[0];
35 Builder::ScaleShiftLayer& Builder::ScaleShiftLayer::setPort(const Port &port) {
36 getLayer()->getOutputPorts()[0] = port;
37 getLayer()->getInputPorts()[0] = port;