diff options
author | Pablo Tello <pablo.tello@arm.com> | 2018-01-23 09:36:04 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:45:00 +0000 |
commit | d6ca478a7e410f8f529c2e505305b46d9fe21a9b (patch) | |
tree | 5c50c06e07f812890f127b1c4933996987f74f17 /arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp | |
parent | d05dce46a14a7b67f322328ecd95bf96bdd30bae (diff) | |
download | ComputeLibrary-d6ca478a7e410f8f529c2e505305b46d9fe21a9b.tar.gz |
COMPMID-784: Added support for biases in WinogradLayer.
1) Updated to the latest code from the RSH repo.
2) Moved winograd transforms into kernels.
3) Added support for biases
Change-Id: I7f39f34a599b49d7d9b549cc10a4f4d4a8007ab8
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/117474
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp')
-rw-r--r-- | arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp b/arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp index 725f6cab65..6a9984a24a 100644 --- a/arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp +++ b/arm_compute/core/NEON/kernels/winograd/direct_convolution.hpp @@ -29,6 +29,7 @@ void direct_convolution( const Tensor4D<Tensor4DShape, float>& input, const Tensor4D<KernelShape, float>& kernel, + const Tensor4D<Tensor4DShape, float>& biases, Tensor4D<Tensor4DShape, float>& output, const PaddingType padding ); |