OpenVINO™ Toolkit
inference-engine
include
builders
ie_tanh_layer.hpp
1
// Copyright (C) 2018-2019 Intel Corporation
2
// SPDX-License-Identifier: Apache-2.0
3
//
4
5
#pragma once
6
7
#include <builders/ie_layer_decorator.hpp>
8
#include <ie_network.hpp>
9
#include <string>
10
11
namespace
InferenceEngine
{
12
namespace
Builder {
13
14
/**
15
* @brief The class represents a builder for TanH layer
16
*/
17
class
INFERENCE_ENGINE_API_CLASS(TanHLayer):
public
LayerDecorator {
18
public
:
19
/**
20
* @brief The constructor creates a builder with the name
21
* @param name Layer name
22
*/
23
explicit
TanHLayer(
const
std::string& name =
""
);
24
/**
25
* @brief The constructor creates a builder from generic builder
26
* @param layer pointer to generic builder
27
*/
28
explicit
TanHLayer(
const
Layer::Ptr& layer);
29
/**
30
* @brief The constructor creates a builder from generic builder
31
* @param layer constant pointer to generic builder
32
*/
33
explicit
TanHLayer(
const
Layer::CPtr& layer);
34
/**
35
* @brief Sets the name for the layer
36
* @param name Layer name
37
* @return reference to layer builder
38
*/
39
TanHLayer& setName(
const
std::string& name);
40
41
/**
42
* @brief Returns port with shapes for the layer
43
* @return Port with shapes
44
*/
45
const
Port& getPort()
const
;
46
/**
47
* @brief Sets port shapes for the layer
48
* @param port Port with shapes
49
* @return reference to layer builder
50
*/
51
TanHLayer& setPort(
const
Port& port);
52
};
53
54
}
// namespace Builder
55
}
// namespace InferenceEngine
InferenceEngine
Definition:
ie_argmax_layer.hpp:11