20.11
InputLayer.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include <
Layer.hpp
>
8
9
namespace
armnn
10
{
11
12
/// A layer user-provided data can be bound to (e.g. inputs, outputs).
13
class
InputLayer
:
public
BindableLayer
14
{
15
public
:
16
/// Makes a workload for the Input type.
17
/// @param [in] graph The graph where this layer can be found.
18
/// @param [in] factory The workload factory which will create the workload.
19
/// @return A pointer to the created workload, or nullptr if not created.
20
virtual
std::unique_ptr<IWorkload>
CreateWorkload
(
const
IWorkloadFactory
& factory)
const override
;
21
22
/// Creates a dynamically-allocated copy of this layer.
23
/// @param [in] graph The graph into which this layer is being cloned.
24
InputLayer
*
Clone
(
Graph
& graph)
const override
;
25
26
/// Check if the input tensor shape(s)
27
/// will lead to a valid configuration of @ref InputLayer.
28
/// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validate.
29
void
ValidateTensorShapesFromInputs
()
override
;
30
31
void
Accept
(
ILayerVisitor
& visitor)
const override
;
32
33
protected
:
34
/// Constructor to create an InputLayer.
35
/// @param id The layer binding id number.
36
/// @param name Optional name for the layer.
37
InputLayer
(
LayerBindingId
id
,
const
char
* name);
38
39
/// Default destructor
40
~InputLayer
() =
default
;
41
};
42
43
}
// namespace
armnn::InputLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Input type.
Definition:
InputLayer.cpp:20
armnn::IWorkloadFactory
Definition:
WorkloadFactory.hpp:21
armnn::BindableLayer
Definition:
Layer.hpp:428
armnn::InputLayer::Accept
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
Definition:
InputLayer.cpp:38
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
armnn::LayerBindingId
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
Definition:
Types.hpp:202
armnn::InputLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of InputLayer.
Definition:
InputLayer.cpp:31
armnn::Graph
Definition:
Graph.hpp:29
armnn::InputLayer::~InputLayer
~InputLayer()=default
Default destructor.
armnn::InputLayer::InputLayer
InputLayer(LayerBindingId id, const char *name)
Constructor to create an InputLayer.
Definition:
InputLayer.cpp:15
Layer.hpp
armnn::InputLayer
A layer user-provided data can be bound to (e.g. inputs, outputs).
Definition:
InputLayer.hpp:13
armnn::InputLayer::Clone
InputLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition:
InputLayer.cpp:26
armnn::ILayerVisitor
Definition:
ILayerVisitor.hpp:16
src
armnn
layers
InputLayer.hpp
Generated on Wed Nov 25 2020 09:59:10 for ArmNN by
1.8.13