ArmNN
 20.02
MultiplicationLayer.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
7 
8 #include "LayerCloneBase.hpp"
9 
10 #include <armnn/TypesUtils.hpp>
13 
14 namespace armnn
15 {
16 
19 {
20 }
21 
22 std::unique_ptr<IWorkload> MultiplicationLayer::CreateWorkload(const IWorkloadFactory& factory) const
23 {
25  return factory.CreateMultiplication(descriptor, PrepInfoAndDesc(descriptor));
26 }
27 
29 {
30  return CloneBase<MultiplicationLayer>(graph, GetName());
31 }
32 
34 {
35  visitor.VisitMultiplicationLayer(this, GetName());
36 }
37 
38 } // namespace armnn
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Multiplication type.
virtual std::unique_ptr< IWorkload > CreateMultiplication(const MultiplicationQueueDescriptor &descriptor, const WorkloadInfo &info) const
Copyright (c) 2020 ARM Limited.
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
virtual void VisitMultiplicationLayer(const IConnectableLayer *layer, const char *name=nullptr)=0
Function that a multiplication layer should call back to when its Accept(ILayerVisitor&) function is ...
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *LayerCreateWorkload.
Definition: Layer.hpp:351
MultiplicationLayer(const char *name)
Constructor to create a MultiplicationLayer.
NOTE: this is an abstract class to encapsulate the element wise operations, it does not implement: st...
const char * GetName() const override
Returns the name of the layer.
Definition: Layer.hpp:305
This layer represents a multiplication operation.
MultiplicationLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.