OpenVINO™ Toolkit
inference-engine
include
builders
ie_lrn_layer.hpp
1
// Copyright (C) 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 LRN layer
16
*/
17
class
INFERENCE_ENGINE_API_CLASS(LRNLayer):
public
LayerDecorator {
18
public
:
19
/**
20
* @brief The constructor creates a builder with the name
21
* @param name Layer name
22
*/
23
explicit
LRNLayer(
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
LRNLayer(
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
LRNLayer(
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
LRNLayer& 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
LRNLayer& setPort(
const
Port& port);
52
/**
53
* @brief Returns side length of the region
54
* @return Size
55
*/
56
size_t
getSize()
const
;
57
/**
58
* @brief Sets side length of the region
59
* @param size Size
60
* @return reference to layer builder
61
*/
62
LRNLayer& setSize(
size_t
size);
63
/**
64
* @brief Returns scaling parameter for the normalizing sum
65
* @return Scaling parameter
66
*/
67
float
getAlpha()
const
;
68
/**
69
* @brief Sets scaling parameter for the normalizing sum
70
* @param alpha Scaling parameter
71
* @return reference to layer builder
72
*/
73
LRNLayer& setAlpha(
float
alpha);
74
/**
75
* @brief Returns exponent for the normalizing sum
76
* @return Exponent
77
*/
78
float
getBeta()
const
;
79
/**
80
* @brief Sets exponent for the normalizing sum
81
* @param beta Exponent
82
* @return reference to layer builder
83
*/
84
LRNLayer& setBeta(
float
beta);
85
/**
86
* @brief Returns region type
87
* @return true if normalizing sum is performed over adjacent channels
88
*/
89
float
getBias()
const
;
90
/**
91
* @brief Sets bias for the normalizing sum
92
* @param bias Bias
93
* @return reference to layer builder
94
*/
95
LRNLayer& setBias(
float
bias);
96
};
97
98
}
// namespace Builder
99
}
// namespace InferenceEngine
InferenceEngine
Definition:
ie_argmax_layer.hpp:11