ArmNN
 24.08
Conv3dImpl.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "BaseIterator.hpp"
9 #include "Decoders.hpp"
10 #include "Encoders.hpp"
11 
12 #include <armnn/Tensor.hpp>
13 
15 
16 namespace armnn
17 {
18 
19 void Convolve3d(const TensorShape& rInputShape,
20  Decoder<float>& rInputDecoder,
21  const TensorShape& rOutputShape,
22  Encoder<float>& rOutputEncoder,
23  const TensorShape& rFilterShape,
24  Decoder<float>& rFilterDecoder,
25  bool biasEnabled,
26  Decoder<float>* pBiasDecoder,
27  DataLayout dataLayout,
28  unsigned int paddingTop,
29  unsigned int paddingLeft,
30  unsigned int paddingFront,
31  unsigned int xStride,
32  unsigned int yStride,
33  unsigned int zStride,
34  unsigned int xDilation,
35  unsigned int yDilation,
36  unsigned int zDilation);
37 
38 } //namespace armnn
armnn::DataLayout
DataLayout
Definition: Types.hpp:62
BaseIterator.hpp
Tensor.hpp
armnn::Convolve3d
void Convolve3d(const TensorShape &rInputShape, Decoder< float > &rInputDecoder, const TensorShape &rOutputShape, Encoder< float > &rOutputEncoder, const TensorShape &rFilterShape, Decoder< float > &rFilterDecoder, bool biasEnabled, Decoder< float > *pBiasDecoder, DataLayout dataLayout, unsigned int paddingTop, unsigned int paddingLeft, unsigned int paddingFront, unsigned int xStride, unsigned int yStride, unsigned int zStride, unsigned int xDilation, unsigned int yDilation, unsigned int zDilation)
Definition: Conv3dImpl.cpp:11
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Encoders.hpp
DataLayoutIndexed.hpp