ie_permute_layer.hpp
1 // Copyright (C) 2018 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include <builders/ie_layer_fragment.hpp>
8 #include <vector>
9 #include <string>
10 
11 namespace InferenceEngine {
12 namespace Builder {
13 
14 /**
15  * @brief The class represents a builder for Permute layer
16  */
17 class INFERENCE_ENGINE_API_CLASS(PermuteLayer): public LayerFragment {
18 public:
19  /**
20  * @brief The constructor creates a builder with the name
21  * @param name Layer name
22  */
23  explicit PermuteLayer(const std::string& name = "");
24  /**
25  * @brief The constructor creates a builder from generic builder
26  * @param genLayer generic builder
27  */
28  explicit PermuteLayer(Layer& genLayer);
29  /**
30  * @brief Sets the name for the layer
31  * @param name Layer name
32  * @return reference to layer builder
33  */
34  PermuteLayer& setName(const std::string& name);
35 
36  /**
37  * @brief Sets weights for layer
38  * @param weights Constant blob with weights
39  * @return reference to layer builder
40  */
41  PermuteLayer& setWeights(const Blob::CPtr& weights);
42  /**
43  * @brief Sets biases for layer
44  * @param biases Constant blob with biases
45  * @return reference to layer builder
46  */
47  PermuteLayer& setBiases(const Blob::CPtr& biases);
48 
49  /**
50  * @brief Returns input port
51  * @return Input port
52  */
53  const Port& getInputPort() const;
54  /**
55  * @brief Sets input port
56  * @param port Input port
57  * @return reference to layer builder
58  */
59  PermuteLayer& setInputPort(const Port& port);
60  /**
61  * @brief Returns output port
62  * @return Output port
63  */
64  const Port& getOutputPort() const;
65  /**
66  * @brief Sets output port
67  * @param port Output port
68  * @return reference to layer builder
69  */
70  PermuteLayer& setOutputPort(const Port& port);
71  /**
72  * @brief Return vector of dimensions indexes for output blob
73  * @return Order of dimensions for output blob
74  */
75  const std::vector<size_t> getOrder() const;
76  /**
77  * @brief Sets the order of dimensions for output blob
78  * @param order dimensions indexes for output blob
79  * @return reference to layer builder
80  */
81  PermuteLayer& setOrder(const std::vector<size_t>& order);
82 };
83 
84 } // namespace Builder
85 } // namespace InferenceEngine
86 
InferenceEngine
Definition: ie_argmax_layer.hpp:11
InferenceEngine::Blob::CPtr
std::shared_ptr< const Blob > CPtr
A smart pointer to the const Blob object.
Definition: ie_blob.h:43