diff options
Diffstat (limited to 'inference-engine/include/builders/ie_scale_shift_layer.hpp')
-rw-r--r-- | inference-engine/include/builders/ie_scale_shift_layer.hpp | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/inference-engine/include/builders/ie_scale_shift_layer.hpp b/inference-engine/include/builders/ie_scale_shift_layer.hpp new file mode 100644 index 000000000..361664e60 --- /dev/null +++ b/inference-engine/include/builders/ie_scale_shift_layer.hpp @@ -0,0 +1,63 @@ +// 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> + +namespace InferenceEngine { +namespace Builder { + +/** + * @brief The class represents a builder for ScaleShift layer + */ +class INFERENCE_ENGINE_API_CLASS(ScaleShiftLayer): public LayerFragment { +public: + /** + * @brief The constructor creates a builder with the name + * @param name Layer name + */ + explicit ScaleShiftLayer(const std::string& name = ""); + /** + * @brief The constructor creates a builder from generic builder + * @param genLayer generic builder + */ + explicit ScaleShiftLayer(Layer& genLayer); + /** + * @brief Sets the name for the layer + * @param name Layer name + * @return reference to layer builder + */ + ScaleShiftLayer& setName(const std::string& name); + + /** + * @brief Returns port with shapes for the layer + * @return Port with shapes + */ + const Port& getPort() const; + /** + * @brief Sets port shapes for the layer + * @param port Port with shapes + * @return reference to layer builder + */ + ScaleShiftLayer& setPort(const Port &port); + + /** + * @brief Sets weights for layer + * @param weights Constant blob with weights + * @return reference to layer builder + */ + ScaleShiftLayer& setWeights(const Blob::CPtr& weights); + /** + * @brief Sets biases for layer + * @param biases Constant blob with biases + * @return reference to layer builder + */ + ScaleShiftLayer& setBiases(const Blob::CPtr& biases); +}; + +} // namespace Builder +} // namespace InferenceEngine |