25.11
Loading...
Searching...
No Matches
Stack.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2017 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include "
Encoders.hpp
"
9
#include "
Decoders.hpp
"
10
11
#include <
armnn/backends/WorkloadData.hpp
>
12
13
namespace
armnn
14
{
15
16
void
Stack
(
const
StackQueueDescriptor
& data,
17
std::vector<std::unique_ptr<
Decoder<float>
>>& inputs,
18
Encoder<float>
& output,
19
const
TensorInfo
& inputInfo,
20
const
TensorInfo
& outputInfo);
21
22
}
// namespace armnn
Decoders.hpp
Encoders.hpp
WorkloadData.hpp
armnn::Decoder
Definition
BaseIterator.hpp:36
armnn::Encoder
Definition
BaseIterator.hpp:51
armnn::TensorInfo
Definition
Tensor.hpp:153
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition
01_00_quick_start.dox:7
armnn::LayerType::Stack
@ Stack
Definition
Types.hpp:496
armnn::StackQueueDescriptor
Definition
WorkloadData.hpp:153
src
backends
reference
workloads
Stack.hpp
Generated on
for Arm NN by
1.14.0