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