ArmNN  NotReleased
PreluLayer.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 "PreluLayer.hpp"
7 
8 #include "LayerCloneBase.hpp"
9 
13 
14 namespace armnn
15 {
16 
17 PreluLayer::PreluLayer(const char* name)
18  : Layer(2, 1, LayerType::Prelu, name)
19 {}
20 
21 std::unique_ptr<IWorkload> PreluLayer::CreateWorkload(const IWorkloadFactory& factory) const
22 {
23  PreluQueueDescriptor descriptor;
24 
25  return factory.CreatePrelu(descriptor, PrepInfoAndDesc(descriptor));
26 }
27 
29 {
30  auto layer = CloneBase<PreluLayer>(graph, GetName());
31 
32  return std::move(layer);
33 }
34 
35 std::vector<TensorShape> PreluLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const
36 {
37  BOOST_ASSERT(inputShapes.size() == 2);
38 
39  const TensorShape& inputShape = inputShapes[0];
40  const TensorShape& alphaShape = inputShapes[1];
41 
42  const unsigned int inputShapeDimensions = inputShape.GetNumDimensions();
43  const unsigned int alphaShapeDimensions = alphaShape.GetNumDimensions();
44 
45  BOOST_ASSERT(inputShapeDimensions > 0);
46  BOOST_ASSERT(alphaShapeDimensions > 0);
47 
48  // The size of the output is the maximum size along each dimension of the input operands,
49  // it starts with the trailing dimensions, and works its way forward
50 
51  unsigned int outputDimensions = std::max(inputShapeDimensions, alphaShapeDimensions);
52 
53  TensorShape outputShape(outputDimensions);
54 
55  int inputShapeIndex = boost::numeric_cast<int>(inputShapeDimensions) - 1;
56  int alphaShapeIndex = boost::numeric_cast<int>(alphaShapeDimensions) - 1;
57  unsigned int outputShapeIndex = outputDimensions - 1;
58 
59  // Loop backwards through the common part of the shapes
60  while (inputShapeIndex >= 0 && alphaShapeIndex >= 0)
61  {
62  unsigned int inputDimension = inputShape[boost::numeric_cast<unsigned int>(inputShapeIndex)];
63  unsigned int alphaDimension = alphaShape[boost::numeric_cast<unsigned int>(alphaShapeIndex)];
64 
65  // Check that the inputs are broadcast compatible
66  BOOST_ASSERT_MSG(inputDimension == alphaDimension || inputDimension == 1 || alphaDimension == 1,
67  "PreluLayer: Dimensions should either match or one should be of size 1");
68 
69  outputShape[outputShapeIndex] = std::max(inputDimension, alphaDimension);
70 
71  inputShapeIndex--;
72  alphaShapeIndex--;
73  outputShapeIndex--;
74  }
75 
76  // Loop backwards through the remaing part of the input shape (if any)
77  while (inputShapeIndex >= 0)
78  {
79  outputShape[outputShapeIndex] = inputShape[boost::numeric_cast<unsigned int>(inputShapeIndex)];
80 
81  inputShapeIndex--;
82  outputShapeIndex--;
83  }
84 
85  // Loop backwards through the remaing part of the alpha shape (if any)
86  while (alphaShapeIndex >= 0)
87  {
88  outputShape[outputShapeIndex] = alphaShape[boost::numeric_cast<unsigned int>(alphaShapeIndex)];
89 
90  alphaShapeIndex--;
91  outputShapeIndex--;
92  }
93 
94  return { outputShape };
95 }
96 
98 {
100 
101  std::vector<TensorShape> inferredShapes = InferOutputShapes(
102  {
105  });
106 
107  BOOST_ASSERT(inferredShapes.size() == 1);
108 
109  ConditionalThrowIfNotEqual<LayerValidationException>(
110  "PreluLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
112  inferredShapes[0]);
113 }
114 
115 void PreluLayer::Accept(ILayerVisitor& visitor) const
116 {
117  visitor.VisitPreluLayer(this, GetName());
118 }
119 
120 } // namespace armnn
PreluLayer(const char *name)
Definition: PreluLayer.cpp:17
const char * GetName() const override
Definition: Layer.hpp:305
unsigned int GetNumDimensions() const
Definition: Tensor.hpp:43
virtual const TensorInfo & GetTensorInfo() const =0
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *LayerCreateWorkload.
Definition: Layer.hpp:344
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Definition: PreluLayer.cpp:21
#define CHECK_LOCATION()
Definition: Exceptions.hpp:169
void ValidateTensorShapesFromInputs() override
Definition: PreluLayer.cpp:97
const IOutputSlot * GetConnection() const override
Definition: Layer.hpp:199
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
Definition: PreluLayer.cpp:35
virtual void VisitPreluLayer(const IConnectableLayer *layer, const char *name=nullptr)=0
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
Definition: Layer.cpp:337
virtual std::unique_ptr< IWorkload > CreatePrelu(const PreluQueueDescriptor &descriptor, const WorkloadInfo &info) const
const TensorShape & GetShape() const
Definition: Tensor.hpp:88
const TensorInfo & GetTensorInfo() const override
Definition: Layer.cpp:63
void Accept(ILayerVisitor &visitor) const override
Definition: PreluLayer.cpp:115
PreluLayer * Clone(Graph &graph) const override
Definition: PreluLayer.cpp:28
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Definition: Layer.hpp:312
const InputSlot & GetInputSlot(unsigned int index) const override
Definition: Layer.hpp:310