← Back to branch summary

~iliaplatone/spacedrone.eu/inova-sis-pack

« back to all changes in this revision

Viewing changes to usr/include/builders/ie_sigmoid_layer.hpp

  • Committer: Ilia Platone
  • Date: 2022-11-15 16:19:28 UTC
  • Revision ID: git-v1:b9f4c8dff67bb705341db6a18f84a3d5f61c23ce
Initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
usr/include/builders/ie_sigmoid_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 Sigmoid layer
 
16
 */
 
17
class INFERENCE_ENGINE_API_CLASS(SigmoidLayer): public LayerDecorator {
 
18
public:
 
19
    /**
 
20
     * @brief The constructor creates a builder with the name
 
21
     * @param name Layer name
 
22
     */
 
23
    explicit SigmoidLayer(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 SigmoidLayer(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 SigmoidLayer(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
    SigmoidLayer& 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
    SigmoidLayer& setPort(const Port& port);
 
52
};
 
53
 
 
54
}  // namespace Builder
 
55
}  // namespace InferenceEngine