13 #include <common/include/ProfilingGuid.hpp> 15 #include <unordered_map> 29 const std::vector<armnn::ConstTensor>& constants,
38 return m_flatBufferBuilder;
53 return m_serializedLayers;
56 flatbuffers::Offset<armnnSerializer::FeatureCompatibilityVersions>
GetVersionTable();
60 flatbuffers::Offset<armnnSerializer::LayerBase> CreateLayerBase(
68 flatbuffers::Offset<armnnSerializer::ConstTensor> CreateConstTensorInfo(
72 flatbuffers::Offset<TensorInfo> CreateTensorInfo(
const armnn::TensorInfo& tensorInfo);
75 flatbuffers::Offset<flatbuffers::Vector<T>> CreateDataVector(
const void* memory,
unsigned int size);
81 std::vector<flatbuffers::Offset<armnnSerializer::InputSlot>> CreateInputSlots(
85 std::vector<flatbuffers::Offset<armnnSerializer::OutputSlot>> CreateOutputSlots(
89 flatbuffers::FlatBufferBuilder m_flatBufferBuilder;
92 std::vector<flatbuffers::Offset<armnnSerializer::AnyLayer>> m_serializedLayers;
95 std::vector<int> m_inputIds;
98 std::vector<int> m_outputIds;
101 std::unordered_map<LayerGuid, uint32_t > m_guidMap;
109 const char* name =
nullptr);
112 const char* name =
nullptr);
116 const char* name =
nullptr);
120 const char* name =
nullptr);
124 const std::vector<armnn::ConstTensor>& constants,
125 const char* name =
nullptr);
128 const char* name =
nullptr);
132 const char* name =
nullptr);
136 const char* name =
nullptr);
140 const char* name =
nullptr);
143 const std::vector<armnn::ConstTensor>& constants,
144 const char* name =
nullptr);
148 const char* name =
nullptr);
152 const char* name =
nullptr);
156 const char* name =
nullptr);
160 const char* name =
nullptr);
163 const char* name =
nullptr);
167 const std::vector<armnn::ConstTensor>& constants,
168 const char* name =
nullptr);
171 const char* name =
nullptr);
175 const char* name =
nullptr);
179 const char* name =
nullptr);
182 const char *name =
nullptr);
186 const char* name =
nullptr);
190 const char* name =
nullptr);
193 const char* name =
nullptr);
197 const char* name =
nullptr);
201 const char* name =
nullptr);
205 const char* name =
nullptr);
209 const char* name =
nullptr);
213 const char* name =
nullptr);
217 const std::vector<armnn::ConstTensor>& constants,
218 const char* name =
nullptr);
225 const char* name =
nullptr);
228 const char* name =
nullptr);
231 const char* name =
nullptr);
234 const char* name =
nullptr);
238 const char* name =
nullptr);
242 const char* name =
nullptr);
246 const char* name =
nullptr);
250 const char* name =
nullptr);
254 const char* name =
nullptr);
257 const char* name =
nullptr);
260 const char* name =
nullptr);
264 const std::vector<armnn::ConstTensor>& constants,
265 const char* name =
nullptr);
268 const std::vector<armnn::ConstTensor>& constants,
269 const char* name =
nullptr);
272 const char* name =
nullptr);
276 const char* name =
nullptr);
280 const char* name =
nullptr);
284 const char* name =
nullptr);
288 const char* name =
nullptr);
292 const char* name =
nullptr);
296 const char* name =
nullptr);
300 const char* name =
nullptr);
304 const char* name =
nullptr);
307 const char* name =
nullptr);
311 const char* name =
nullptr);
315 const char* name =
nullptr);
319 const char* name =
nullptr);
323 const char* name =
nullptr);
326 const char* name =
nullptr);
329 const char* name =
nullptr);
333 const std::vector<armnn::ConstTensor>& constants,
334 const char* =
nullptr);
338 const char* name =
nullptr);
342 const std::vector<armnn::ConstTensor>& constants,
343 const char* name =
nullptr);
361 bool SaveSerializedToStream(std::ostream& stream);
A ViewsDescriptor for the SplitterLayer.
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
flatbuffers::FlatBufferBuilder & GetFlatBufferBuilder()
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.
A ReshapeDescriptor for the ReshapeLayer.
A ComparisonDescriptor for the ComparisonLayer.
A Convolution2dDescriptor for the Convolution2dLayer.
Main network class which provides the interface for building up a neural network. ...
A LogicalBinaryDescriptor for the LogicalBinaryLayer.
A SpaceToDepthDescriptor for the SpaceToDepthLayer.
A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer.
void ExecuteStrategy(const armnn::IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id) override
int LayerBindingId
Type of identifiers for bindable layers (inputs, outputs).
A ResizeBilinearDescriptor for the ResizeBilinearLayer.
Base class for all descriptors.
A StackDescriptor for the StackLayer.
A PadDescriptor for the PadLayer.
An LstmDescriptor for the LstmLayer.
A L2NormalizationDescriptor for the L2NormalizationLayer.
An ArgMinMaxDescriptor for ArgMinMaxLayer.
An OriginsDescriptor for the ConcatLayer.
A ReduceDescriptor for the REDUCE operators.
A FullyConnectedDescriptor for the FullyConnectedLayer.
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
A GatherDescriptor for the GatherLayer.
A StandInDescriptor for the StandIn layer.
A QLstmDescriptor for the QLstmLayer.
std::vector< int > & GetInputIds()
An ActivationDescriptor for the ActivationLayer.
A SliceDescriptor for the SliceLayer.
A Convolution3dDescriptor for the Convolution3dLayer.
A Pooling3dDescriptor for the Pooling3dLayer.
A SpaceToBatchNdDescriptor for the SpaceToBatchNdLayer.
arm::pipe::ProfilingGuid LayerGuid
Define LayerGuid type.
std::vector< int > & GetOutputIds()
A ElementwiseUnaryDescriptor for the ElementwiseUnaryLayer.
A MeanDescriptor for the MeanLayer.
A TransposeDescriptor for the TransposeLayer.
A StridedSliceDescriptor for the StridedSliceLayer.
A Pooling2dDescriptor for the Pooling2dLayer.
A NormalizationDescriptor for the NormalizationLayer.
std::vector< flatbuffers::Offset< armnnSerializer::AnyLayer > > & GetSerializedLayers()
flatbuffers::Offset< armnnSerializer::FeatureCompatibilityVersions > GetVersionTable()
An InstanceNormalizationDescriptor for InstanceNormalizationLayer.
A ChannelShuffleDescriptor for the ChannelShuffle operator.
A SoftmaxDescriptor for the SoftmaxLayer.
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
A FillDescriptor for the FillLayer.
A BatchNormalizationDescriptor for the BatchNormalizationLayer.
A PermuteDescriptor for the PermuteLayer.