blob: 94ef1cc9ffb28b8cd63ffb779048c3081ef54bf8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright (C) 2018 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//
#pragma once
#include <builders/ie_layer_fragment.hpp>
#include <ie_inetwork.hpp>
#include <string>
namespace InferenceEngine {
namespace Builder {
/**
* @brief The class represents a builder for Power layer
*/
class INFERENCE_ENGINE_API_CLASS(PowerLayer): public LayerFragment {
public:
/**
* @brief The constructor creates a builder with the name
* @param name Layer name
*/
explicit PowerLayer(const std::string& name = "");
/**
* @brief The constructor creates a builder from generic builder
* @param genLayer generic builder
*/
explicit PowerLayer(Layer& genLayer);
/**
* @brief Sets the name for the layer
* @param name Layer name
* @return reference to layer builder
*/
PowerLayer& setName(const std::string& name);
/**
* @brief Returns port with shapes for the layer
* @return Port with shapes
*/
const Port& getPort() const;
/**
* @brief Sets port shapes for the layer
* @param port Port with shapes
* @return reference to layer builder
*/
PowerLayer& setPort(const Port& port);
/**
* @brief Returns power
* @return Power parameter
*/
float getPower() const;
/**
* @brief Sets the power parameter
* @param power Power parameter
* @return reference to layer builder
*/
PowerLayer& setPower(float power);
/**
* @brief Returns scaling parameter
* @return Scaling
*/
float getScale() const;
/**
* @brief Sets scaling parameter
* @param scale Scaling parameter
* @return reference to layer builder
*/
PowerLayer& setScale(float scale);
/**
* @brief Returns shifting parameter
* @return Shift
*/
float getShift() const;
/**
* @brief Sets shift for the layer
* @param shift Shifting parameter
* @return reference to layer builder
*/
PowerLayer& setShift(float shift);
};
} // namespace Builder
} // namespace InferenceEngine
|
