diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2018-05-30 11:44:26 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:53:09 +0000 |
| commit | 7df27869aff38b07b50e4fe589f6b2cf51954a92 (patch) | |
| tree | f3e0fc514b6d90306de49dea28ad42a1144cb185 /tests/validation/fixtures/WinogradConvolutionLayerFixture.h | |
| parent | c084f0d4d2ee94bedc31b5e04c2936c91cecf883 (diff) | |
| download | ComputeLibrary-7df27869aff38b07b50e4fe589f6b2cf51954a92.tar.gz | |
COMPMID-1162: Enable NHWC data layout support for NEWinogradConvolutionLayer - part1
In this first part we reworked the configuration of the kernels as before we
passed the raw pointer to the buffer within the configuration of the function
Change-Id: I83d3cb64c562303093c7f0ae52395ecd080a5d52
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/133560
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'tests/validation/fixtures/WinogradConvolutionLayerFixture.h')
| -rw-r--r-- | tests/validation/fixtures/WinogradConvolutionLayerFixture.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h index 6381b99131..f40f3d2e43 100644 --- a/tests/validation/fixtures/WinogradConvolutionLayerFixture.h +++ b/tests/validation/fixtures/WinogradConvolutionLayerFixture.h @@ -55,7 +55,8 @@ class WinogradConvolutionLayerValidationFixture : public framework::Fixture { public: template <typename...> - void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, DataType data_type, ActivationLayerInfo act_info) + void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrideInfo info, Size2D dilation, + DataType data_type, ActivationLayerInfo act_info) { ARM_COMPUTE_UNUSED(dilation); @@ -84,7 +85,7 @@ protected: } } - TensorType compute_target(const TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, const PadStrideInfo &info, + TensorType compute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const PadStrideInfo &info, DataType data_type, ActivationLayerInfo act_info) { // Create tensors |