inference-engine/include/builders/ie_reshape_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 <ie_inetwork.hpp>
   9 #include <string>
  10 #include <vector>
  11
  12 namespace InferenceEngine {
  13 namespace Builder {
  14
  15 /**
  16  * @brief The class represents a builder for Reshape layer
  17  */
  18 class INFERENCE_ENGINE_API_CLASS(ReshapeLayer): public LayerFragment {
  19 public:
  20     /**
  21      * @brief The constructor creates a builder with the name
  22      * @param name Layer name
  23      */
  24     explicit ReshapeLayer(const std::string& name = "");
  25     /**
  26      * @brief The constructor creates a builder from generic builder
  27      * @param genLayer generic builder
  28      */
  29     explicit ReshapeLayer(Layer& genLayer);
  30     /**
  31      * @brief Sets the name for the layer
  32      * @param name Layer name
  33      * @return reference to layer builder
  34      */
  35     ReshapeLayer& setName(const std::string& name);
  36
  37     /**
  38      * @brief Returns input port
  39      * @return Input port
  40      */
  41     const Port& getInputPort() const;
  42     /**
  43      * @brief Sets input port
  44      * @param port Input port
  45      * @return reference to layer builder
  46      */
  47     ReshapeLayer& setInputPort(const Port& port);
  48     /**
  49      * @brief Returns output port
  50      * @return Output port
  51      */
  52     const Port& getOutputPort() const;
  53     /**
  54      * @brief Sets output port
  55      * @param port Output port
  56      * @return reference to layer builder
  57      */
  58     ReshapeLayer& setOutputPort(const Port& port);
  59     /**
  60      * @brief Returns reshape dimensions
  61      * @return Dimensions
  62      */
  63     const std::vector<int> getDims() const;
  64     /**
  65      * @brief Sets reshape dimensions
  66      * @param dims Dimensions
  67      * @return reference to layer builder
  68      */
  69     ReshapeLayer& setDims(const std::vector<int>& dims);
  70 };
  71
  72 }  // namespace Builder
  73 }  // namespace InferenceEngine