summaryrefslogtreecommitdiff
path: root/inference-engine/include/builders/ie_sigmoid_layer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/include/builders/ie_sigmoid_layer.hpp')
-rw-r--r--inference-engine/include/builders/ie_sigmoid_layer.hpp50
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