10 using namespace armnn;
15 for (
auto&& layer : layers)
17 for (
auto&& it = layer->BeginInputSlots(); it != layer->EndInputSlots(); ++it)
19 result.push_back(&(*it));
28 for (
auto && layer : layers)
30 for (
auto&& it = layer->BeginOutputSlots(); it != layer->EndOutputSlots(); ++it)
32 result.push_back(&(*it));
42 return std::make_unique<SubgraphView>(std::move(inputs), std::move(outputs), std::move(layers));
48 auto backendFactory = backendRegistry.GetFactory(backendId);
49 auto backendObjPtr = backendFactory();
55 unsigned int channels,
60 using namespace armnn;
64 return TensorShape{ batches, channels, height, width };
66 return TensorShape{ batches, height, width, channels };
std::unique_ptr< IBackendInternal > IBackendInternalUniquePtr
std::list< Layer * > Layers
std::vector< InputSlot * > InputSlots
BackendRegistry & BackendRegistryInstance()
armnn::TensorShape MakeTensorShape(unsigned int batches, unsigned int channels, unsigned int height, unsigned int width, armnn::DataLayout layout)
armnn::IBackendInternalUniquePtr CreateBackendObject(const armnn::BackendId &backendId)
SubgraphView::InputSlots CreateInputsFrom(const std::vector< Layer *> &layers)
std::unique_ptr< SubgraphView > SubgraphViewPtr
SubgraphView::OutputSlots CreateOutputsFrom(const std::vector< Layer *> &layers)
constexpr const char * GetDataLayoutName(DataLayout dataLayout)
std::vector< OutputSlot * > OutputSlots
SubgraphView::SubgraphViewPtr CreateSubgraphViewFrom(SubgraphView::InputSlots &&inputs, SubgraphView::OutputSlots &&outputs, SubgraphView::Layers &&layers)