diff options
Diffstat (limited to 'src/armnn/layers/ResizeBilinearLayer.cpp')
-rw-r--r-- | src/armnn/layers/ResizeBilinearLayer.cpp | 74 |
1 files changed, 0 insertions, 74 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 |