1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
5 #include <builders/ie_tanh_layer.hpp>
9 using namespace InferenceEngine;
11 Builder::TanHLayer::TanHLayer(const std::string& name): LayerDecorator("TanH", name) {
12 getLayer()->getOutputPorts().resize(1);
13 getLayer()->getInputPorts().resize(1);
16 Builder::TanHLayer::TanHLayer(const Layer::Ptr& layer): LayerDecorator(layer) {
20 Builder::TanHLayer::TanHLayer(const Layer::CPtr& layer): LayerDecorator(layer) {
24 Builder::TanHLayer& Builder::TanHLayer::setName(const std::string& name) {
25 getLayer()->setName(name);
29 const Port& Builder::TanHLayer::getPort() const {
30 return getLayer()->getOutputPorts()[0];
33 Builder::TanHLayer& Builder::TanHLayer::setPort(const Port &port) {
34 getLayer()->getOutputPorts()[0] = port;
35 getLayer()->getInputPorts()[0] = port;
39 REG_VALIDATOR_FOR(TanH, [] (const InferenceEngine::Builder::Layer::CPtr& input_layer, bool partial) {
40 if (!input_layer->getInputPorts().empty() &&
41 !input_layer->getOutputPorts().empty() &&
42 !input_layer->getInputPorts()[0].shape().empty() &&
43 !input_layer->getOutputPorts()[0].shape().empty() &&
44 input_layer->getInputPorts()[0].shape() != input_layer->getOutputPorts()[0].shape()) {
45 THROW_IE_EXCEPTION << "Input and output ports should be equal";