ArmNN
 25.02
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
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 
10 namespace armnn
11 {
12 
14 {
15 public:
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< 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
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &, TensorShape const &, unsigned int const *) 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.