ie_power_layer.hpp
1 // Copyright (C) 2018 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include <builders/ie_layer_fragment.hpp>
8 #include <ie_inetwork.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 LayerFragment {
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 genLayer generic builder
27  */
28  explicit PowerLayer(Layer& genLayer);
29  /**
30  * @brief Sets the name for the layer
31  * @param name Layer name
32  * @return reference to layer builder
33  */
34  PowerLayer& setName(const std::string& name);
35 
36  /**
37  * @brief Returns port with shapes for the layer
38  * @return Port with shapes
39  */
40  const Port& getPort() const;
41  /**
42  * @brief Sets port shapes for the layer
43  * @param port Port with shapes
44  * @return reference to layer builder
45  */
46  PowerLayer& setPort(const Port& port);
47  /**
48  * @brief Returns power
49  * @return Power parameter
50  */
51  float getPower() const;
52  /**
53  * @brief Sets the power parameter
54  * @param power Power parameter
55  * @return reference to layer builder
56  */
57  PowerLayer& setPower(float power);
58  /**
59  * @brief Returns scaling parameter
60  * @return Scaling
61  */
62  float getScale() const;
63  /**
64  * @brief Sets scaling parameter
65  * @param scale Scaling parameter
66  * @return reference to layer builder
67  */
68  PowerLayer& setScale(float scale);
69  /**
70  * @brief Returns shifting parameter
71  * @return Shift
72  */
73  float getShift() const;
74  /**
75  * @brief Sets shift for the layer
76  * @param shift Shifting parameter
77  * @return reference to layer builder
78  */
79  PowerLayer& setShift(float shift);
80 };
81 
82 } // namespace Builder
83 } // namespace InferenceEngine
InferenceEngine
Definition: ie_argmax_layer.hpp:11
ie_inetwork.hpp
a header file for the Inference Engine Network interface