ArmNN
 24.02
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
armnn::WorkloadFactoryBase::CreateTensorHandle
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, const bool) const override
Definition: WorkloadFactoryBase.hpp:25
armnn::WorkloadFactoryBase
Definition: WorkloadFactoryBase.hpp:13
armnn::DataLayout
DataLayout
Definition: Types.hpp:62
armnn::TensorInfo
Definition: Tensor.hpp:152
armnn::ITensorHandle
Definition: ITensorHandle.hpp:16
WorkloadFactory.hpp
armnn::TensorShape
Definition: Tensor.hpp:20
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22
armnn::QueueDescriptor
Definition: WorkloadData.hpp:24
armnn::WorkloadFactoryBase::CreateWorkload
std::unique_ptr< IWorkload > CreateWorkload(LayerType, const QueueDescriptor &, const WorkloadInfo &) const override
Backends should implement their own CreateWorkload function with a switch statement.
Definition: WorkloadFactoryBase.hpp:34
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::WorkloadFactoryBase::CreateTensorHandle
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &, DataLayout, const bool) const override
Definition: WorkloadFactoryBase.hpp:29
armnn::WorkloadFactoryBase::SupportsSubTensors
bool SupportsSubTensors() const override
Definition: WorkloadFactoryBase.hpp:16
armnn::LayerType
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below.
Definition: Types.hpp:491
armnn::WorkloadFactoryBase::CreateSubTensorHandle
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &, TensorShape const &, unsigned int const *) const override
Definition: WorkloadFactoryBase.hpp:19