From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_batch_to_space_nd_layer_8cpp_source.xhtml | 159 +++++++++++++++++++++++ 1 file changed, 159 insertions(+) create mode 100644 20.02/_batch_to_space_nd_layer_8cpp_source.xhtml (limited to '20.02/_batch_to_space_nd_layer_8cpp_source.xhtml') diff --git a/20.02/_batch_to_space_nd_layer_8cpp_source.xhtml b/20.02/_batch_to_space_nd_layer_8cpp_source.xhtml new file mode 100644 index 0000000000..4dbad60008 --- /dev/null +++ b/20.02/_batch_to_space_nd_layer_8cpp_source.xhtml @@ -0,0 +1,159 @@ + + + + + + + + + + + + + +ArmNN: src/armnn/layers/BatchToSpaceNdLayer.cpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
BatchToSpaceNdLayer.cpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #include "BatchToSpaceNdLayer.hpp"
7 #include "LayerCloneBase.hpp"
8 #include "LayerWithParameters.hpp"
9 #include "BatchToSpaceNdLayer.hpp"
10 
11 #include <armnn/TypesUtils.hpp>
12 
13 #include <armnnUtils/DataLayoutIndexed.hpp>
14 
15 #include <backendsCommon/CpuTensorHandle.hpp>
16 #include <backendsCommon/WorkloadData.hpp>
17 #include <backendsCommon/WorkloadFactory.hpp>
18 
19 #include <numeric>
20 
21 using namespace armnnUtils;
22 
23 namespace armnn
24 {
25 
26 BatchToSpaceNdLayer::BatchToSpaceNdLayer(const armnn::BatchToSpaceNdDescriptor& param, const char* name)
27  : LayerWithParameters(1, 1, LayerType::BatchToSpaceNd, param, name)
28 {
29 }
30 
31 std::unique_ptr<IWorkload> BatchToSpaceNdLayer::CreateWorkload(const IWorkloadFactory& factory) const
32 {
33  BatchToSpaceNdQueueDescriptor descriptor;
34 
35  return factory.CreateBatchToSpaceNd(descriptor, PrepInfoAndDesc(descriptor));
36 }
37 
38 BatchToSpaceNdLayer* BatchToSpaceNdLayer::Clone(Graph& graph) const
39 {
40  auto layer = CloneBase<BatchToSpaceNdLayer>(graph, m_Param, GetName());
41  return std::move(layer);
42 }
43 
44 void BatchToSpaceNdLayer::ValidateTensorShapesFromInputs()
45 {
46  VerifyLayerConnections(1, CHECK_LOCATION());
47 
48  auto inferredShapes = InferOutputShapes({GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape()});
49 
50  BOOST_ASSERT(inferredShapes.size() == 1);
51 
52  ConditionalThrowIfNotEqual<LayerValidationException>(
53  "BatchToSpaceLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
54  GetOutputSlot(0).GetTensorInfo().GetShape(),inferredShapes[0]);
55 }
56 
57 std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const
58 {
59  BOOST_ASSERT(inputShapes.size() == 1);
60 
61  const TensorShape& inputShape = inputShapes[0];
62  TensorShape outputShape(inputShape);
63 
64  unsigned int accumulatedBlockShape = std::accumulate(m_Param.m_BlockShape.begin(),
65  m_Param.m_BlockShape.end(),
66  1U,
67  std::multiplies<>());
68 
69  BOOST_ASSERT(inputShape[0] % accumulatedBlockShape == 0);
70 
71  outputShape[0] = inputShape[0] / accumulatedBlockShape;
72 
73  DataLayoutIndexed dimensionIndices = m_Param.m_DataLayout;
74  unsigned int heightIndex = dimensionIndices.GetHeightIndex();
75  unsigned int widthIndex = dimensionIndices.GetWidthIndex();
76 
77  unsigned int heightCrop = m_Param.m_Crops[0].first + m_Param.m_Crops[0].second;
78  unsigned int widthCrop = m_Param.m_Crops[1].first + m_Param.m_Crops[1].second;
79 
80  unsigned int outputHeight = inputShape[heightIndex] * m_Param.m_BlockShape[0];
81  unsigned int outputWidth = inputShape[widthIndex] * m_Param.m_BlockShape[1];
82 
83  BOOST_ASSERT_MSG(heightCrop <= outputHeight,
84  "BatchToSpaceLayer: Overall height crop should be less than or equal to the uncropped output height.");
85 
86  BOOST_ASSERT_MSG(widthCrop <= outputWidth,
87  "BatchToSpaceLayer: Overall width crop should be less than or equal to the uncropped output width.");
88 
89  outputShape[heightIndex] = outputHeight - heightCrop;
90  outputShape[widthIndex] = outputWidth - widthCrop;
91 
92  return std::vector<TensorShape>({ outputShape });
93 }
94 
95 void BatchToSpaceNdLayer::Accept(ILayerVisitor& visitor) const
96 {
97  visitor.VisitBatchToSpaceNdLayer(this, GetParameters(), GetName());
98 }
99 
100 } // namespace armnn
LayerWithParameters.hpp
+
WorkloadData.hpp
+
DataLayoutIndexed.hpp
+
armnn::IWorkloadFactory::CreateBatchToSpaceNd
virtual std::unique_ptr< IWorkload > CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &Info) const
Definition: WorkloadFactory.cpp:1123
+
armnn::LayerWithParameters< BatchToSpaceNdDescriptor >::m_Param
BatchToSpaceNdDescriptor m_Param
The parameters for the layer (not including tensor-valued weights etc.).
Definition: LayerWithParameters.hpp:50
+
armnn::LayerWithParameters< BatchToSpaceNdDescriptor >::GetParameters
const BatchToSpaceNdDescriptor & GetParameters() const
Definition: LayerWithParameters.hpp:18
+
armnn::BatchToSpaceNdLayer::InferOutputShapes
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const 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.
Definition: BatchToSpaceNdLayer.cpp:57
+
armnnUtils::DataLayoutIndexed::GetWidthIndex
unsigned int GetWidthIndex() const
Definition: DataLayoutIndexed.hpp:25
+
armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:88
+
WorkloadFactory.hpp
+
armnn::BatchToSpaceNdLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the BatchToSpaceNd type.
Definition: BatchToSpaceNdLayer.cpp:31
+ +
armnn::BatchToSpaceNdDescriptor::m_DataLayout
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
Definition: Descriptors.hpp:684
+
armnn::BatchToSpaceNdLayer::BatchToSpaceNdLayer
BatchToSpaceNdLayer(const BatchToSpaceNdDescriptor &param, const char *name)
Constructor to create a BatchToSpaceNdLayer.
Definition: BatchToSpaceNdLayer.cpp:26
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+
BatchToSpaceNdLayer.hpp
+ +
armnn::InputSlot::GetConnection
const IOutputSlot * GetConnection() const override
Definition: Layer.hpp:199
+ +
armnn::BatchToSpaceNdDescriptor
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
Definition: Descriptors.hpp:657
+
armnnUtils::DataLayoutIndexed::GetHeightIndex
unsigned int GetHeightIndex() const
Definition: DataLayoutIndexed.hpp:24
+
armnn::Layer::VerifyLayerConnections
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
Definition: Layer.cpp:338
+
armnn::Layer::GetInputSlot
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
Definition: Layer.hpp:310
+
TypesUtils.hpp
+
armnn::BatchToSpaceNdLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of BatchToSpaceNdLayer.
Definition: BatchToSpaceNdLayer.cpp:44
+
armnn::BatchToSpaceNdLayer::Accept
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
Definition: BatchToSpaceNdLayer.cpp:95
+
armnn::BatchToSpaceNdDescriptor::m_BlockShape
std::vector< unsigned int > m_BlockShape
Block shape values.
Definition: Descriptors.hpp:680
+
armnnUtils::DataLayoutIndexed
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout...
Definition: DataLayoutIndexed.hpp:17
+
armnn::BatchToSpaceNdLayer
This layer represents a BatchToSpaceNd operation.
Definition: BatchToSpaceNdLayer.hpp:13
+
CHECK_LOCATION
#define CHECK_LOCATION()
Definition: Exceptions.hpp:192
+ +
armnn::ILayerVisitor::VisitBatchToSpaceNdLayer
virtual void VisitBatchToSpaceNdLayer(const IConnectableLayer *layer, const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr)=0
Function that a batch to space ND layer should call back to when its Accept(ILayerVisitor&) function ...
+
armnn::BatchToSpaceNdLayer::Clone
BatchToSpaceNdLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: BatchToSpaceNdLayer.cpp:38
+
CpuTensorHandle.hpp
+
armnn::BatchToSpaceNdDescriptor::m_Crops
std::vector< std::pair< unsigned int, unsigned int > > m_Crops
The values to crop from the input dimension.
Definition: Descriptors.hpp:682
+
armnn::LayerWithParameters< BatchToSpaceNdDescriptor >::PrepInfoAndDesc
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *LayerCreateWorkload.
Definition: LayerWithParameters.hpp:43
+ + +
armnn::Layer::GetOutputSlot
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
Definition: Layer.hpp:312
+
armnn::IOutputSlot::GetTensorInfo
virtual const TensorInfo & GetTensorInfo() const =0
+
armnn::Layer::GetName
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:305
+
armnn::BatchToSpaceNd
void BatchToSpaceNd(const DataLayoutIndexed &dataLayout, const TensorInfo &inputTensorInfo, const TensorInfo &outputTensorInfo, const std::vector< unsigned int > &blockShape, const std::vector< std::pair< unsigned int, unsigned int >> &cropsData, Decoder< float > &inputDecoder, Encoder< float > &outputEncoder)
Definition: BatchToSpaceNd.cpp:35
+ +
armnn::OutputSlot::GetTensorInfo
const TensorInfo & GetTensorInfo() const override
Definition: Layer.cpp:63
+
LayerCloneBase.hpp
+ +
+
+ + + + -- cgit v1.2.1