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 14 #include <armnnUtils/DataLayoutIndexed.hpp> 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 39