ArmNN
 24.08
StandInLayer.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2019 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 
10 namespace armnn
11 {
12 
13 /// This layer represents an unknown operation in the input graph.
14 class StandInLayer : public LayerWithParameters<StandInDescriptor>
15 {
16 public:
17  /// Empty implementation explictly does NOT create a workload. Throws Exception if called.
18  /// @param [in] graph The graph where this layer can be found.
19  /// @param [in] factory The workload factory which will create the workload.
20  /// @return Does not return anything. Throws Exception if called.
21  virtual std::unique_ptr<IWorkload>CreateWorkload(const IWorkloadFactory& factory) const override;
22 
23  /// Creates a dynamically-allocated copy of this layer.
24  /// @param [in] graph The graph into which this layer is being cloned.
25  StandInLayer* Clone(Graph& graph) const override;
26 
27  /// Check if the input tensor shape(s)
28  /// Does nothing since cannot validate any properties of this layer.
29  /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated.
30  void ValidateTensorShapesFromInputs() override;
31 
32  /// Empty implementation that throws Exception if called.
33  /// otherwise infers the output shapes from given input shapes and layer properties.
34  /// @param [in] inputShapes The input shapes layer has.
35  /// @return Does not return anything. Throws Exception if called.
36  std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
37 
38  void ExecuteStrategy(IStrategy& strategy) const override;
39 
40 
41 protected:
42  /// Constructor to create a StandInLayer.
43  /// @param [in] param StandInDescriptor to configure the stand-in operation.
44  /// @param [in] name Optional name for the layer.
45  StandInLayer(const StandInDescriptor& param, const char* name);
46 
47  /// Default destructor
48  ~StandInLayer() = default;
49 };
50 
51 } //namespace armnn
52 
53 
54 
55 
armnn::IStrategy
Definition: IStrategy.hpp:16
armnn::LayerWithParameters
Definition: LayerWithParameters.hpp:14
armnn::StandInLayer::StandInLayer
StandInLayer(const StandInDescriptor &param, const char *name)
Constructor to create a StandInLayer.
Definition: StandInLayer.cpp:12
armnn::StandInLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) Does nothing since cannot validate any properties of this layer.
Definition: StandInLayer.cpp:37
armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22
armnn::StandInLayer::InferOutputShapes
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
Empty implementation that throws Exception if called.
Definition: StandInLayer.cpp:31
armnn::StandInLayer
This layer represents an unknown operation in the input graph.
Definition: StandInLayer.hpp:14
LayerWithParameters.hpp
armnn::StandInLayer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: StandInLayer.cpp:44
armnn::StandInLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Empty implementation explictly does NOT create a workload.
Definition: StandInLayer.cpp:17
armnn::StandInDescriptor
A StandInDescriptor for the StandIn layer.
Definition: Descriptors.hpp:1281
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::StandInLayer::~StandInLayer
~StandInLayer()=default
Default destructor.
armnn::StandInLayer::Clone
StandInLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: StandInLayer.cpp:26
armnn::Graph
Definition: Graph.hpp:30