ArmNN
 25.11
Loading...
Searching...
No Matches
WorkloadFactoryBase.hpp
Go to the documentation of this file.
1//
2// Copyright © 2019, 2023 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
9
10namespace armnn
11{
12
14{
15public:
16 bool SupportsSubTensors() const override
17 { return false; };
18
19 std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& /*parent*/,
20 TensorShape const& /*subTensorShape*/,
21 unsigned int const */*subTensorOrigin*/) const override
22 { return nullptr; };
23
24
25 std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& /*tensorInfo*/,
26 const bool /*IsMemoryManaged*/) const override
27 { return nullptr; }
28
29 std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& /*tensorInfo*/,
30 DataLayout /*dataLayout*/,
31 const bool /*IsMemoryManaged*/) const override
32 { return nullptr; }
33
34 std::unique_ptr<IWorkload> CreateWorkload(LayerType /*type*/,
35 const QueueDescriptor& /*descriptor*/,
36 const WorkloadInfo& /*info*/) const override
37 { return nullptr; }
38
39};
40
41} //namespace armnn
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &, TensorShape const &, unsigned int const *) const override
std::unique_ptr< IWorkload > CreateWorkload(LayerType, const QueueDescriptor &, const WorkloadInfo &) const override
Backends should implement their own CreateWorkload function with a switch statement.
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, const bool) const override
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, DataLayout, const bool) const override
bool SupportsSubTensors() const override
Copyright (c) 2021 ARM Limited and Contributors.
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.
Definition Types.hpp:494
DataLayout
Definition Types.hpp:63
Contains information about TensorInfos of a layer.