ie_batch_normalization_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  * @deprecated Use ngraph API instead.
16  * @brief The class represents a builder for BatchNormalization layer
17  */
18 class INFERENCE_ENGINE_NN_BUILDER_API_CLASS(BatchNormalizationLayer): public LayerDecorator {
19 public:
20  /**
21  * @brief The constructor creates a builder with the name
22  * @param name Layer name
23  */
24  explicit BatchNormalizationLayer(const std::string& name = "");
25  /**
26  * @brief The constructor creates a builder from generic builder
27  * @param layer pointer to generic builder
28  */
29  explicit BatchNormalizationLayer(const Layer::Ptr& layer);
30  /**
31  * @brief The constructor creates a builder from generic builder
32  * @param layer constant pointer to generic builder
33  */
34  explicit BatchNormalizationLayer(const Layer::CPtr& layer);
35  /**
36  * @brief Sets the name for the layer
37  * @param name Layer name
38  * @return reference to layer builder
39  */
40  BatchNormalizationLayer& setName(const std::string& name);
41 
42  /**
43  * @brief Returns port with shapes for the layer
44  * @return Port with shapes
45  */
46  const Port& getPort() const;
47  /**
48  * @brief Sets port shapes for the layer
49  * @param port Port with shapes
50  * @return reference to layer builder
51  */
52  BatchNormalizationLayer& setPort(const Port& port);
53 
54  /**
55  * @brief Returns epsilon
56  * @return Epsilon
57  */
58  float getEpsilon() const;
59  /**
60  * @brief Sets epsilon
61  * @param eps Epsilon
62  * @return reference to layer builder
63  */
64  BatchNormalizationLayer& setEpsilon(float eps);
65 };
66 
67 } // namespace Builder
68 } // namespace InferenceEngine
InferenceEngine
Inference Engine API.
Definition: ie_argmax_layer.hpp:11