20.02
WorkloadInfo.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020 Arm Ltd. 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 inputs and outputs to a layer.
15
/// This is needed at construction of workloads, but are not stored.
16
struct
WorkloadInfo
17
{
18
std::vector<TensorInfo>
m_InputTensorInfos
;
19
std::vector<TensorInfo>
m_OutputTensorInfos
;
20
};
21
22
}
//namespace armnn
Tensor.hpp
armnn
Copyright (c) 2020 ARM Limited.
Definition:
00_introduction.dox:25
armnn::WorkloadInfo::m_InputTensorInfos
std::vector< TensorInfo > m_InputTensorInfos
Definition:
WorkloadInfo.hpp:18
armnn::WorkloadInfo::m_OutputTensorInfos
std::vector< TensorInfo > m_OutputTensorInfos
Definition:
WorkloadInfo.hpp:19
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition:
WorkloadInfo.hpp:16
include
armnn
backends
WorkloadInfo.hpp
Generated on Fri Mar 13 2020 16:09:07 for ArmNN by
1.8.13