// // Copyright © 2020 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once #include namespace armnn { /// This layer represents a memory copy operation. class UnmapLayer : public Layer { public: /// Makes a workload for the Unmap type. /// @param [in] graph The graph where this layer can be found. /// @param [in] factory The workload factory which will create the workload. /// @return A pointer to the created workload, or nullptr if not created. virtual std::unique_ptrCreateWorkload(const IWorkloadFactory& factory) const override; /// Creates a dynamically-allocated copy of this layer. /// @param [in] graph The graph into which this layer is being cloned. UnmapLayer* Clone(Graph& graph) const override; /// Check if the input tensor shape(s) /// will lead to a valid configuration of @ref UnmapLayer. /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated. void ValidateTensorShapesFromInputs() override; ARMNN_NO_DEPRECATE_WARN_BEGIN void Accept(ILayerVisitor& visitor) const override; ARMNN_NO_DEPRECATE_WARN_END protected: /// Constructor to create a UnmapLayer. /// @param [in] name Optional name for the layer. UnmapLayer(const char* name); /// Default destructor ~UnmapLayer() = default; }; } // namespace