1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
7 #include <builders/ie_layer_decorator.hpp>
8 #include <ie_network.hpp>
11 namespace InferenceEngine {
15 * @brief The class represents a builder for MVN layer
17 class INFERENCE_ENGINE_API_CLASS(MVNLayer): public LayerDecorator {
20 * @brief The constructor creates a builder with the name
21 * @param name Layer name
23 explicit MVNLayer(const std::string& name = "");
25 * @brief The constructor creates a builder from generic builder
26 * @param layer pointer to generic builder
28 explicit MVNLayer(const Layer::Ptr& layer);
30 * @brief The constructor creates a builder from generic builder
31 * @param layer constant pointer to generic builder
33 explicit MVNLayer(const Layer::CPtr& layer);
35 * @brief Sets the name for the layer
36 * @param name Layer name
37 * @return reference to layer builder
39 MVNLayer& setName(const std::string& name);
42 * @brief Returns port with shapes for the layer
43 * @return Port with shapes
45 const Port& getPort() const;
47 * @brief Sets port shapes for the layer
48 * @param port Port with shapes
49 * @return reference to layer builder
51 MVNLayer& setPort(const Port& port);
53 * @brief Returns across channels value
54 * @return true if mean values are shared across channels
56 bool getAcrossChannels() const;
58 * @brief Sets across channels
59 * @param flag true if mean values are shared across channels
60 * @return reference to layer builder
62 MVNLayer& setAcrossChannels(bool flag);
64 * @brief Returns normalize variance
65 * @return true if variance normalization is performed
67 bool getNormalize() const;
69 * @brief Sets normalize variance
70 * @param flag true if variance normalization is performed
71 * @return reference to layer builder
73 MVNLayer& setNormalize(bool flag);
75 * @brief Return epsilon
78 float getEpsilon() const;
82 * @return reference to layer builder
84 MVNLayer& setEpsilon(float eps);
87 } // namespace Builder
88 } // namespace InferenceEngine