ArmNN
 20.02
ActivationLayer.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "ActivationLayer.hpp"
6 #include "LayerCloneBase.hpp"
7 
8 #include <armnn/TypesUtils.hpp>
11 
12 namespace armnn
13 {
14 
16  : LayerWithParameters(1, 1, LayerType::Activation, param, name)
17 {
18 }
19 
20 std::unique_ptr<IWorkload> ActivationLayer::CreateWorkload(const IWorkloadFactory& factory) const
21 {
22  ActivationQueueDescriptor descriptor;
23  return factory.CreateActivation(descriptor, PrepInfoAndDesc(descriptor));
24 }
25 
27 {
28  return CloneBase<ActivationLayer>(graph, m_Param, GetName());
29 }
30 
32 {
34 
35  auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });
36 
37  BOOST_ASSERT(inferredShapes.size() == 1);
38 
39  ConditionalThrowIfNotEqual<LayerValidationException>(
40  "ActivationLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
42  inferredShapes[0]);
43 }
44 
46 {
47  visitor.VisitActivationLayer(this, GetParameters(), GetName());
48 }
49 
50 } // namespace armnn
ActivationDescriptor m_Param
The parameters for the layer (not including tensor-valued weights etc.).
const ActivationDescriptor & GetParameters() const
const TensorShape & GetShape() const
Definition: Tensor.hpp:88
virtual void VisitActivationLayer(const IConnectableLayer *layer, const ActivationDescriptor &activationDescriptor, const char *name=nullptr)=0
Function that an activation layer should call back to when its Accept(ILayerVisitor&) function is inv...
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
Infer the shape of the output(s) based on the provided input shape(s)
Definition: Layer.cpp:371
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of ActivationLayer.
This layer represents an activation operation with the specified activation function.
Copyright (c) 2020 ARM Limited.
virtual std::unique_ptr< IWorkload > CreateActivation(const ActivationQueueDescriptor &descriptor, const WorkloadInfo &info) const
const IOutputSlot * GetConnection() const override
Definition: Layer.hpp:199
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
Definition: Layer.cpp:338
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
Definition: Layer.hpp:310
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
float Activation(float in, ActivationFunction function, float a, float b)
Definition: Activation.cpp:13
ActivationLayer(const ActivationDescriptor &param, const char *name)
Constructor to create an ActivationLayer.
An ActivationDescriptor for the ActivationLayer.
Definition: Descriptors.hpp:20
#define CHECK_LOCATION()
Definition: Exceptions.hpp:192
ActivationLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *LayerCreateWorkload.
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
Definition: Layer.hpp:312
virtual const TensorInfo & GetTensorInfo() const =0
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:305
const TensorInfo & GetTensorInfo() const override
Definition: Layer.cpp:63
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Activation type.