ArmNN
 25.11
Loading...
Searching...
No Matches
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
14namespace armnn
15{
16
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
Copyright (c) 2021 ARM Limited and Contributors.
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)
A TransposeConvolution2dDescriptor for the TransposeConvolution2dLayer.