ArmNN  NotReleased
InputLayer.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <Layer.hpp>
8 
9 namespace armnn
10 {
11 
13 class InputLayer : public BindableLayer
14 {
15 public:
20  virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override;
21 
24  InputLayer* Clone(Graph& graph) const override;
25 
28  void ValidateTensorShapesFromInputs() override;
29 
30  void Accept(ILayerVisitor& visitor) const override;
31 
32 protected:
36  InputLayer(LayerBindingId id, const char* name);
37 
39  ~InputLayer() = default;
40 };
41 
42 } // namespace
armnn::InputLayer
A layer user-provided data can be bound to (e.g. inputs, outputs).
Definition: InputLayer.hpp:13
armnn::InputLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Definition: InputLayer.cpp:31
armnn::InputLayer::~InputLayer
~InputLayer()=default
Default destructor.
armnn::InputLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Definition: InputLayer.cpp:20
armnn::InputLayer::Accept
void Accept(ILayerVisitor &visitor) const override
Definition: InputLayer.cpp:38
Layer.hpp
armnn::InputLayer::InputLayer
InputLayer(LayerBindingId id, const char *name)
Definition: InputLayer.cpp:15
armnn::LayerBindingId
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
Definition: Types.hpp:168
armnn::InputLayer::Clone
InputLayer * Clone(Graph &graph) const override
Definition: InputLayer.cpp:26