forked from Qortal/Brooklyn
26 lines
627 B
C++
26 lines
627 B
C++
//
|
|
// Copyright © 2020 Arm Ltd. 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();
|
|
};
|
|
|
|
} //namespace armnn
|