ArmNN
 24.02
WorkloadInfo.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2020-2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <armnn/Tensor.hpp>
8 
9 #include <vector>
10 
11 namespace armnn
12 {
13 
14 /// Contains information about TensorInfos of a layer.
15 /// This is needed at construction of workloads, but are not stored.
17 {
18  std::vector<TensorInfo> m_InputTensorInfos;
19  std::vector<TensorInfo> m_OutputTensorInfos;
20  std::string m_Name;
24 };
25 
26 struct MemoryInfo
27 {
28  unsigned int m_OutputSlotIndex;
29  size_t m_Size{ 0 };
30  size_t m_Alignment{ 64 };
31 };
32 
34 {
36 };
37 
38 } //namespace armnn
armnn::WorkloadInfo::m_Name
std::string m_Name
Definition: WorkloadInfo.hpp:20
armnn::Optional
Definition: Optional.hpp:270
armnn::MemoryInfo::m_Alignment
size_t m_Alignment
Definition: WorkloadInfo.hpp:30
armnn::MemoryInfo::m_OutputSlotIndex
unsigned int m_OutputSlotIndex
Definition: WorkloadInfo.hpp:28
armnn::MemoryRequirements::m_IntraLayerTensors
armnn::Optional< std::vector< MemoryInfo > > m_IntraLayerTensors
Definition: WorkloadInfo.hpp:35
armnn::MemoryRequirements
Definition: WorkloadInfo.hpp:33
armnn::WorkloadInfo::m_ConvolutionMethod
Optional< std::string > m_ConvolutionMethod
Definition: WorkloadInfo.hpp:23
armnn::WorkloadInfo::m_OutputTensorInfos
std::vector< TensorInfo > m_OutputTensorInfos
Definition: WorkloadInfo.hpp:19
armnn::WorkloadInfo::m_WeightsTensorInfo
Optional< TensorInfo > m_WeightsTensorInfo
Definition: WorkloadInfo.hpp:21
armnn::WorkloadInfo::m_BiasTensorInfo
Optional< TensorInfo > m_BiasTensorInfo
Definition: WorkloadInfo.hpp:22
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::EmptyOptional
EmptyOptional is used to initialize the Optional class in case we want to have default value for an O...
Definition: Optional.hpp:32
armnn::MemoryInfo
Definition: WorkloadInfo.hpp:26
armnn::MemoryInfo::m_Size
size_t m_Size
Definition: WorkloadInfo.hpp:29
Tensor.hpp
armnn::WorkloadInfo::m_InputTensorInfos
std::vector< TensorInfo > m_InputTensorInfos
Definition: WorkloadInfo.hpp:18
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6