← Back to branch summary

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

« back to all changes in this revision

Viewing changes to usr/include/builders/ie_concat_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_concat_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
#include <vector>
 
11
 
 
12
namespace InferenceEngine {
 
13
namespace Builder {
 
14
 
 
15
/**
 
16
 * @brief The class represents a builder for Concat layer
 
17
 */
 
18
class INFERENCE_ENGINE_API_CLASS(ConcatLayer): public LayerDecorator {
 
19
public:
 
20
    /**
 
21
     * @brief The constructor creates a builder with the name
 
22
     * @param name Layer name
 
23
     */
 
24
    explicit ConcatLayer(const std::string& name = "");
 
25
    /**
 
26
     * @brief The constructor creates a builder from generic builder
 
27
     * @param layer pointer to generic builder
 
28
     */
 
29
    explicit ConcatLayer(const Layer::Ptr& layer);
 
30
    /**
 
31
     * @brief The constructor creates a builder from generic builder
 
32
     * @param layer constant pointer to generic builder
 
33
     */
 
34
    explicit ConcatLayer(const Layer::CPtr& layer);
 
35
    /**
 
36
     * @brief Sets the name for the layer
 
37
     * @param name Layer name
 
38
     * @return reference to layer builder
 
39
     */
 
40
    ConcatLayer& setName(const std::string& name);
 
41
 
 
42
    /**
 
43
     * @brief Returns vector with input ports
 
44
     * @return vector with ports
 
45
     */
 
46
    const std::vector<Port>& getInputPorts() const;
 
47
    /**
 
48
     * @brief Sets input ports
 
49
     * @param ports Vector of input ports
 
50
     * @return reference to layer builder
 
51
     */
 
52
    ConcatLayer& setInputPorts(const std::vector<Port>& ports);
 
53
    /**
 
54
     * @brief Returns output port
 
55
     * @return Output port
 
56
     */
 
57
    const Port& getOutputPort() const;
 
58
    /**
 
59
     * @brief Sets output port
 
60
     * @param port Output port
 
61
     * @return reference to layer builder
 
62
     */
 
63
    ConcatLayer& setOutputPort(const Port& port);
 
64
    /**
 
65
     * @brief Returns axis
 
66
     * @return Axis
 
67
     */
 
68
    size_t getAxis() const;
 
69
    /**
 
70
     * @brief Sets axis
 
71
     * @param axis Axis
 
72
     * @return reference to layer builder
 
73
     */
 
74
    ConcatLayer& setAxis(size_t axis);
 
75
 
 
76
private:
 
77
    size_t axis = 1;
 
78
};
 
79
 
 
80
}  // namespace Builder
 
81
}  // namespace InferenceEngine