ie_batch_normalization_layer.hpp
Go to the documentation of this file.
1 // Copyright (C) 2018-2020 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 /**
6  * @file
7  */
8 
9 #pragma once
10 
12 #include <ie_network.hpp>
13 #include <string>
14 
15 namespace InferenceEngine {
16 namespace Builder {
17 
18 /**
19  * @deprecated Use ngraph API instead.
20  * @brief The class represents a builder for BatchNormalization layer
21  */
22 IE_SUPPRESS_DEPRECATED_START
23 class INFERENCE_ENGINE_NN_BUILDER_API_CLASS(BatchNormalizationLayer): public LayerDecorator {
24 public:
25  /**
26  * @brief The constructor creates a builder with the name
27  * @param name Layer name
28  */
29  explicit BatchNormalizationLayer(const std::string& name = "");
30  /**
31  * @brief The constructor creates a builder from generic builder
32  * @param layer pointer to generic builder
33  */
34  explicit BatchNormalizationLayer(const Layer::Ptr& layer);
35  /**
36  * @brief The constructor creates a builder from generic builder
37  * @param layer constant pointer to generic builder
38  */
39  explicit BatchNormalizationLayer(const Layer::CPtr& layer);
40  /**
41  * @brief Sets the name for the layer
42  * @param name Layer name
43  * @return reference to layer builder
44  */
45  BatchNormalizationLayer& setName(const std::string& name);
46 
47  /**
48  * @brief Returns port with shapes for the layer
49  * @return Port with shapes
50  */
51  const Port& getPort() const;
52  /**
53  * @brief Sets port shapes for the layer
54  * @param port Port with shapes
55  * @return reference to layer builder
56  */
57  BatchNormalizationLayer& setPort(const Port& port);
58 
59  /**
60  * @brief Returns epsilon
61  * @return Epsilon
62  */
63  float getEpsilon() const;
64  /**
65  * @brief Sets epsilon
66  * @param eps Epsilon
67  * @return reference to layer builder
68  */
69  BatchNormalizationLayer& setEpsilon(float eps);
70 };
71 IE_SUPPRESS_DEPRECATED_END
72 
73 } // namespace Builder
74 } // namespace InferenceEngine
Inference Engine API.
Definition: ie_argmax_layer.hpp:15
std::shared_ptr< const Layer > CPtr
A shared pointer to the constant Layer builder.
Definition: ie_layer_builder.hpp:52
std::string name
Layer name.
Definition: ie_layers.h:42
This class defines the basic functional for layer builders.
Definition: ie_layer_decorator.hpp:26
std::shared_ptr< Layer > Ptr
A shared pointer to the Layer builder.
Definition: ie_layer_builder.hpp:48
The class represents a builder for BatchNormalization layer.
Definition: ie_batch_normalization_layer.hpp:23
A header file for the Inference Engine Network interface.
This class is the main object to describe the Inference Engine port.
Definition: ie_network.hpp:230