diff options
Diffstat (limited to 'src/armnn/layers')
-rw-r--r-- | src/armnn/layers/ResizeBilinearLayer.cpp | 74 | ||||
-rw-r--r-- | src/armnn/layers/ResizeBilinearLayer.hpp | 49 |
2 files changed, 0 insertions, 123 deletions
diff --git a/src/armnn/layers/ResizeBilinearLayer.cpp b/src/armnn/layers/ResizeBilinearLayer.cpp deleted file mode 100644 index 03fe317e79..0000000000 --- a/src/armnn/layers/ResizeBilinearLayer.cpp +++ /dev/null @@ -1,74 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#include "ResizeBilinearLayer.hpp" - -#include "LayerCloneBase.hpp" - -#include <armnn/TypesUtils.hpp> - -#include <backendsCommon/WorkloadData.hpp> -#include <backendsCommon/WorkloadFactory.hpp> - -#include <DataLayoutIndexed.hpp> - -using namespace armnnUtils; - -namespace armnn -{ - -ResizeBilinearLayer::ResizeBilinearLayer(const ResizeBilinearDescriptor& param, const char* name) - : LayerWithParameters(1, 1, LayerType::ResizeBilinear, param, name) -{ -} - -std::unique_ptr<IWorkload> ResizeBilinearLayer::CreateWorkload(const Graph& graph, - const IWorkloadFactory& factory) const -{ - ResizeBilinearQueueDescriptor descriptor; - return factory.CreateResizeBilinear(descriptor, PrepInfoAndDesc(descriptor, graph)); -} - -ResizeBilinearLayer* ResizeBilinearLayer::Clone(Graph& graph) const -{ - return CloneBase<ResizeBilinearLayer>(graph, m_Param, GetName()); -} - -std::vector<TensorShape> ResizeBilinearLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const -{ - BOOST_ASSERT(inputShapes.size() == 1); - const TensorShape& inputShape = inputShapes[0]; - const DataLayoutIndexed dimensionIndices = m_Param.m_DataLayout; - unsigned int outWidth = m_Param.m_TargetWidth; - unsigned int outHeight = m_Param.m_TargetHeight; - unsigned int outChannels = inputShape[dimensionIndices.GetChannelsIndex()]; - unsigned int outBatch = inputShape[0]; - - TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? - TensorShape( { outBatch, outHeight, outWidth, outChannels } ) : - TensorShape( { outBatch, outChannels, outHeight, outWidth }); - - return std::vector<TensorShape>({ tensorShape }); -} - -void ResizeBilinearLayer::ValidateTensorShapesFromInputs() -{ - VerifyLayerConnections(1, CHECK_LOCATION()); - - auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() }); - - BOOST_ASSERT(inferredShapes.size() == 1); - - ConditionalThrowIfNotEqual<LayerValidationException>( - "ResizeBilinearLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.", - GetOutputSlot(0).GetTensorInfo().GetShape(), - inferredShapes[0]); -} - -void ResizeBilinearLayer::Accept(ILayerVisitor& visitor) const -{ - visitor.VisitResizeBilinearLayer(this, GetParameters(), GetName()); -} - -} // namespace armnn diff --git a/src/armnn/layers/ResizeBilinearLayer.hpp b/src/armnn/layers/ResizeBilinearLayer.hpp deleted file mode 100644 index 4bf264cb45..0000000000 --- a/src/armnn/layers/ResizeBilinearLayer.hpp +++ /dev/null @@ -1,49 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "LayerWithParameters.hpp" - -namespace armnn -{ - -/// This layer represents a resize bilinear operation. -class ResizeBilinearLayer : public LayerWithParameters<ResizeBilinearDescriptor> -{ -public: - /// Makes a workload for the ResizeBilinear 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_ptr<IWorkload>CreateWorkload(const Graph& graph, - 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. - ResizeBilinearLayer* Clone(Graph& graph) const override; - - /// Check if the input tensor shape(s) - /// will lead to a valid configuration of @ref ResizeBilinearLayer. - void ValidateTensorShapesFromInputs() override; - - /// By default returns inputShapes if the number of inputs are equal to number of outputs, - /// otherwise infers the output shapes from given input shapes and layer properties. - /// @param [in] inputShapes The input shapes layer has. - /// @return A vector to the inferred output shape. - std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; - - void Accept(ILayerVisitor& visitor) const override; - -protected: - /// Constructor to create a ResizeBilinearLayerLayer. - /// @param [in] param ResizeBilinearDescriptor to configure the resize bilinear operation. - /// @param [in] name Optional name for the layer. - ResizeBilinearLayer(const ResizeBilinearDescriptor& param, const char* name); - - /// Default destructor - ~ResizeBilinearLayer() = default; -}; - -} // namespace |