//
// 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