blob: fdc905fea2835e246741078a8af72b8a320b08ee (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
//
// Copyright © 2019 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "StandInLayer.hpp"
#include "LayerCloneBase.hpp"
namespace armnn
{
StandInLayer::StandInLayer(const StandInDescriptor& param, const char* name)
: LayerWithParameters(param.m_NumInputs, 1, LayerType::StandIn, param, name)
{
}
std::unique_ptr<IWorkload> StandInLayer::CreateWorkload(const Graph& graph, const IWorkloadFactory& factory) const
{
// This throws in the event that it's called. We would expect that any backend that
// "claims" to support the StandInLayer type would actually substitute it with a PrecompiledLayer
// during graph optimization. There is no interface on the IWorkloadFactory to create a StandInWorkload.
throw Exception("Stand in layer does not support creating workloads");
}
StandInLayer* StandInLayer::Clone(Graph& graph) const
{
return CloneBase<StandInLayer>(graph, m_Param, GetName());
}
std::vector<TensorShape> StandInLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const
{
throw Exception("Stand in layer does not support infering output shapes");
}
void StandInLayer::ValidateTensorShapesFromInputs()
{
// Cannot validate this layer since no implementation details can be known by the framework
// so do nothing here.
}
void StandInLayer::Accept(ILayerVisitor& visitor) const
{
visitor.VisitStandInLayer(this, GetParameters(), GetName());
}
} //namespace armnn
|