diff options
Diffstat (limited to 'inference-engine/include/builders/ie_sigmoid_layer.hpp')
-rw-r--r-- | inference-engine/include/builders/ie_sigmoid_layer.hpp | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/inference-engine/include/builders/ie_sigmoid_layer.hpp b/inference-engine/include/builders/ie_sigmoid_layer.hpp new file mode 100644 index 000000000..6c483588a --- /dev/null +++ b/inference-engine/include/builders/ie_sigmoid_layer.hpp @@ -0,0 +1,50 @@ +// Copyright (C) 2018 Intel Corporation +// SPDX-License-Identifier: Apache-2.0 +// + +#pragma once + +#include <builders/ie_layer_fragment.hpp> +#include <ie_inetwork.hpp> +#include <string> + +namespace InferenceEngine { +namespace Builder { + +/** + * @brief The class represents a builder for Sigmoid layer + */ +class INFERENCE_ENGINE_API_CLASS(SigmoidLayer): public LayerFragment { +public: + /** + * @brief The constructor creates a builder with the name + * @param name Layer name + */ + explicit SigmoidLayer(const std::string& name = ""); + /** + * @brief The constructor creates a builder from generic builder + * @param genLayer generic builder + */ + explicit SigmoidLayer(Layer& genLayer); + /** + * @brief Sets the name for the layer + * @param name Layer name + * @return reference to layer builder + */ + SigmoidLayer& setName(const std::string& name); + + /** + * @brief Returns port with shapes for the layer + * @return Port with shapes + */ + const Port& getPort() const; + /** + * @brief Sets port shapes for the layer + * @param port Port with shapes + * @return reference to layer builder + */ + SigmoidLayer& setPort(const Port& port); +}; + +} // namespace Builder +} // namespace InferenceEngine |