Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / include / builders / ie_batch_normalization_layer.hpp

diff --git a/inference-engine/include/builders/ie_batch_normalization_layer.hpp b/inference-engine/include/builders/ie_batch_normalization_layer.hpp
index dbdf538..14d0fe2 100644 (file)
--- a/inference-engine/include/builders/ie_batch_normalization_layer.hpp
+++ b/inference-engine/include/builders/ie_batch_normalization_layer.hpp
@@ -1,11 +1,11 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
 #pragma once
 
-#include <builders/ie_layer_fragment.hpp>
-#include <ie_inetwork.hpp>
+#include <builders/ie_layer_decorator.hpp>
+#include <ie_network.hpp>
 #include <string>
 
 namespace InferenceEngine {
@@ -14,7 +14,7 @@ namespace Builder {
 /**
  * @brief The class represents a builder for BatchNormalization layer
  */
-class INFERENCE_ENGINE_API_CLASS(BatchNormalizationLayer): public LayerFragment {
+class INFERENCE_ENGINE_API_CLASS(BatchNormalizationLayer): public LayerDecorator {
 public:
     /**
      * @brief The constructor creates a builder with the name
@@ -23,9 +23,14 @@ public:
     explicit BatchNormalizationLayer(const std::string& name = "");
     /**
      * @brief The constructor creates a builder from generic builder
-     * @param genLayer generic builder
+     * @param layer pointer to generic builder
      */
-    explicit BatchNormalizationLayer(Layer& genLayer);
+    explicit BatchNormalizationLayer(const Layer::Ptr& layer);
+    /**
+     * @brief The constructor creates a builder from generic builder
+     * @param layer constant pointer to generic builder
+     */
+    explicit BatchNormalizationLayer(const Layer::CPtr& layer);
     /**
      * @brief Sets the name for the layer
      * @param name Layer name
@@ -46,19 +51,6 @@ public:
     BatchNormalizationLayer& setPort(const Port &port);
 
     /**
-     * @brief Sets weights for layer
-     * @param weights Constant blob with weights
-     * @return reference to layer builder
-     */
-    BatchNormalizationLayer& setWeights(const Blob::CPtr& weights);
-    /**
-     * @brief Sets biases for layer
-     * @param biases Constant blob with biases
-     * @return reference to layer builder
-     */
-    BatchNormalizationLayer& setBiases(const Blob::CPtr& biases);
-
-    /**
      * @brief Returns epsilon
      * @return Epsilon
      */
@@ -69,12 +61,6 @@ public:
      * @return reference to layer builder
      */
     BatchNormalizationLayer& setEpsilon(float eps);
-
-    /**
-     * @brief Validates layer before creation
-     * @param layer generic layer builder
-     */
-    static void validate(const Layer& layer);
 };
 
 }  // namespace Builder