OpenVINO™ Toolkit
inference-engine
include
builders
ie_power_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 Power layer
16
*/
17
class
INFERENCE_ENGINE_API_CLASS(PowerLayer):
public
LayerDecorator {
18
public
:
19
/**
20
* @brief The constructor creates a builder with the name
21
* @param name Layer name
22
*/
23
explicit
PowerLayer(
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
PowerLayer(
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
PowerLayer(
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
PowerLayer& 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
PowerLayer& setPort(
const
Port& port);
52
/**
53
* @brief Returns power
54
* @return Power parameter
55
*/
56
float
getPower()
const
;
57
/**
58
* @brief Sets the power parameter
59
* @param power Power parameter
60
* @return reference to layer builder
61
*/
62
PowerLayer& setPower(
float
power);
63
/**
64
* @brief Returns scaling parameter
65
* @return Scaling
66
*/
67
float
getScale()
const
;
68
/**
69
* @brief Sets scaling parameter
70
* @param scale Scaling parameter
71
* @return reference to layer builder
72
*/
73
PowerLayer& setScale(
float
scale);
74
/**
75
* @brief Returns shifting parameter
76
* @return Shift
77
*/
78
float
getShift()
const
;
79
/**
80
* @brief Sets shift for the layer
81
* @param shift Shifting parameter
82
* @return reference to layer builder
83
*/
84
PowerLayer& setShift(
float
shift);
85
};
86
87
}
// namespace Builder
88
}
// namespace InferenceEngine
InferenceEngine
Definition:
ie_argmax_layer.hpp:11