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