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/_transpose_as_reshape_8hpp_source.xhtml | 138 ++++++++++++++++++++++++++ 1 file changed, 138 insertions(+) create mode 100644 20.02/_transpose_as_reshape_8hpp_source.xhtml (limited to '20.02/_transpose_as_reshape_8hpp_source.xhtml') diff --git a/20.02/_transpose_as_reshape_8hpp_source.xhtml b/20.02/_transpose_as_reshape_8hpp_source.xhtml new file mode 100644 index 0000000000..7fa688d58f --- /dev/null +++ b/20.02/_transpose_as_reshape_8hpp_source.xhtml @@ -0,0 +1,138 @@ + + + + + + + + + + + + + +ArmNN: src/armnn/optimizations/TransposeAsReshape.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
TransposeAsReshape.hpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2020 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include "Optimization.hpp"
8 
9 namespace armnn
10 {
11 namespace optimizations
12 {
13 
14 class TransposeAsReshapeImpl
15 {
16 public:
17  /// Run for every TransposeLayer. Replaces it with a ReshapeLayer if they are equivalent.
18  void Run(Graph& graph, TransposeLayer& transpose) const
19  {
20  if (IsReshape(transpose))
21  {
22  const TensorInfo& outInfo = transpose.GetOutputHandler().GetTensorInfo();
23 
24  const std::string name = std::string("as_reshape-") + transpose.GetName();
25  const ReshapeDescriptor descriptor{outInfo.GetShape()};
26  // Inserts NewLayer so layers don't need to be re-sorted.
27  auto reshape = graph.InsertNewLayer<ReshapeLayer>(transpose.GetInputSlot(0), descriptor, name.c_str());
28  reshape->GetOutputHandler().SetTensorInfo(outInfo);
29 
30  // Bypass transpose. It will be deleted since it's left unconnected.
31  transpose.GetOutputSlot().MoveAllConnections(reshape->GetOutputSlot());
32  }
33  }
34 
35 protected:
36  TransposeAsReshapeImpl() = default;
37  ~TransposeAsReshapeImpl() = default;
38 
39 private:
40  static bool IsReshape(const TransposeLayer& layer)
41  {
42  const TensorShape& outShape = layer.GetOutputHandler().GetTensorInfo().GetShape();
43  const PermutationVector& permutation = layer.GetPermutation();
44 
45  const unsigned int numDimensions = permutation.GetSize();
46  std::map<unsigned int, unsigned int> permuteMappings;
47  for (unsigned int i = 0; i < permutation.GetSize(); ++i)
48  {
49  permuteMappings[permutation[i]] = i;
50  }
51 
52  std::vector<unsigned int> permuteVector;
53  for (unsigned int i = 0; i < permutation.GetSize(); ++i)
54  {
55  permuteVector.push_back(permuteMappings.at(i));
56  }
57 
58  unsigned int lastGtOne = 0;
59  while ((lastGtOne < numDimensions) && (outShape[(permuteVector[lastGtOne])] == 1U))
60  {
61  ++lastGtOne;
62  }
63 
64  bool isReshape = true;
65  for (unsigned int i = lastGtOne + 1U; isReshape && (i < numDimensions); ++i)
66  {
67  if (outShape[permuteVector[i]] > 1U)
68  {
69  isReshape = permuteVector[lastGtOne] < permuteVector[i];
70  lastGtOne = i;
71  }
72  }
73 
74  return isReshape;
75  }
76 };
77 
78 using TransposeAsReshape = OptimizeForType<TransposeLayer, TransposeAsReshapeImpl>;
79 
80 } // namespace optimizations
81 } // namespace armnn
+
armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:88
+
armnn::ReshapeDescriptor
A ReshapeDescriptor for the ReshapeLayer.
Definition: Descriptors.hpp:758
+ + + +
armnn::ReshapeLayer
This layer represents a reshape operation.
Definition: ReshapeLayer.hpp:13
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+ +
armnn::PermutationVector::GetSize
SizeType GetSize() const
Definition: Types.hpp:202
+
armnn::Layer::GetInputSlot
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
Definition: Layer.hpp:310
+ + +
armnn::TransposeLayer
This layer represents a transpose operation.
Definition: TransposeLayer.hpp:13
+
armnn::OutputHandler::SetTensorInfo
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.
Definition: OutputHandler.cpp:17
+ +
armnn::Layer::GetOutputHandler
const OutputHandler & GetOutputHandler(unsigned int i=0) const
Definition: Layer.hpp:221
+
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::Layer::GetName
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:305
+
Optimization.hpp
+
armnn::Graph::InsertNewLayer
LayerT * InsertNewLayer(InputSlot &insertBefore, Args &&... args)
Inserts a new layer between the output slot currently connected to insertBefore and insertBefore itse...
Definition: Graph.hpp:409
+
armnn::TransposeLayer::GetPermutation
const PermutationVector & GetPermutation() const
Definition: TransposeLayer.hpp:35
+
armnn::OutputSlot::MoveAllConnections
void MoveAllConnections(OutputSlot &destination)
Moves all connections to another OutputSlot.
Definition: Layer.cpp:112
+
armnn::optimizations::TransposeAsReshapeImpl::Run
void Run(Graph &graph, TransposeLayer &transpose) const
Run for every TransposeLayer. Replaces it with a ReshapeLayer if they are equivalent.
Definition: TransposeAsReshape.hpp:18
+
armnn::OutputHandler::GetTensorInfo
const TensorInfo & GetTensorInfo() const
Gets the matching TensorInfo for the output.
Definition: OutputHandler.hpp:44
+
+
+ + + + -- cgit v1.2.1