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 /tests | |
| 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 'tests')
| -rw-r--r-- | tests/validation/fixtures/WinogradLayerFixture.h | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/validation/fixtures/WinogradLayerFixture.h b/tests/validation/fixtures/WinogradLayerFixture.h index 7aa26c7143..d7f0cbfdf5 100644 --- a/tests/validation/fixtures/WinogradLayerFixture.h +++ b/tests/validation/fixtures/WinogradLayerFixture.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -88,7 +88,7 @@ protected: // Create and configure function FunctionType conv; - conv.configure(&src, &weights, nullptr, &dst, info); + conv.configure(&src, &weights, &bias, &dst, info); ARM_COMPUTE_EXPECT(src.info()->is_resizable(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT(weights.info()->is_resizable(), framework::LogLevel::ERRORS); @@ -98,8 +98,8 @@ protected: // Allocate tensors src.allocator()->allocate(); weights.allocator()->allocate(); - bias.allocator()->allocate(); dst.allocator()->allocate(); + bias.allocator()->allocate(); ARM_COMPUTE_EXPECT(!src.info()->is_resizable(), framework::LogLevel::ERRORS); ARM_COMPUTE_EXPECT(!weights.info()->is_resizable(), framework::LogLevel::ERRORS); @@ -109,7 +109,7 @@ protected: // Fill tensors fill(AccessorType(src), 0, -1.f, 1.f); fill(AccessorType(weights), 1, -1.f, 1.f); - fill(AccessorType(bias), 2, 0.f, 0.f); + fill(AccessorType(bias), 2, -1.f, 1.f); fill(AccessorType(dst), 3, -1.f, 1.f); // Compute NEWinogradLayer function @@ -128,7 +128,7 @@ protected: // Fill reference fill(src, 0, -1.f, 1.f); fill(weights, 1, -1.f, 1.f); - fill(bias, 2, 0.f, 0.f); + fill(bias, 2, -1.f, 1.f); return reference::convolution_layer<T>(src, weights, bias, output_shape, info); } |