ArmNN
 24.02
TransposeConvolution2d.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 "Decoders.hpp"
9 #include "Encoders.hpp"
10 
11 #include <armnn/Descriptors.hpp>
12 #include <armnn/Tensor.hpp>
13 
14 namespace armnn
15 {
16 
17 void TransposeConvolution2dImpl(const TransposeConvolution2dDescriptor& descriptor,
18  const TensorShape& inputShape,
19  Decoder<float>& inputDecoder,
20  const TensorShape& outputShape,
21  Encoder<float>& outputEncoder,
22  const TensorShape& weightsShape,
23  Decoder<float>& weightsDecoder,
24  Decoder<float>* biasesDecoder);
25 
26 } // namespace armnn
armnn::TransposeConvolution2dImpl
void TransposeConvolution2dImpl(const TransposeConvolution2dDescriptor &descriptor, const TensorShape &inputShape, Decoder< float > &inputDecoder, const TensorShape &outputShape, Encoder< float > &outputEncoder, const TensorShape &weightsShape, Decoder< float > &weightsDecoder, Decoder< float > *biasesDecoder)
Definition: TransposeConvolution2d.cpp:15
Descriptors.hpp
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp