WorkloadInfo.hpp

Go to the documentation of this file.

//
// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
 
#include <armnn/Tensor.hpp>
 
#include <vector>
 
namespace armnn
{
 
/// Contains information about TensorInfos of a layer.
/// This is needed at construction of workloads, but are not stored.
struct WorkloadInfo
{
    std::vector<TensorInfo> m_InputTensorInfos;
    std::vector<TensorInfo> m_OutputTensorInfos;
    Optional<TensorInfo> m_WeightsTensorInfo = EmptyOptional();
    Optional<TensorInfo> m_BiasTensorInfo = EmptyOptional();
    Optional<std::string> m_ConvolutionMethod = EmptyOptional();
};
 
struct MemoryInfo
{
    unsigned int   m_OutputSlotIndex;
    size_t         m_Size{ 0 };
    size_t         m_Alignment{ 64 };
};
 
struct MemoryRequirements
{
    armnn::Optional<std::vector<MemoryInfo>> m_IntraLayerTensors;
};
 
} //namespace armnn