diff options
Diffstat (limited to 'tests/validation/fixtures/DirectConvolutionLayerFixture.h')
-rw-r--r-- | tests/validation/fixtures/DirectConvolutionLayerFixture.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/validation/fixtures/DirectConvolutionLayerFixture.h b/tests/validation/fixtures/DirectConvolutionLayerFixture.h index d21a2e7ec3..614aa20753 100644 --- a/tests/validation/fixtures/DirectConvolutionLayerFixture.h +++ b/tests/validation/fixtures/DirectConvolutionLayerFixture.h @@ -162,6 +162,9 @@ protected: TensorType bias = create_tensor<TensorType>(bias_shape, bias_data_type, 1, quantization_info); TensorType dst = create_tensor<TensorType>(output_shape, data_type, 1, quantization_info, data_layout); + add_padding_x({ &src, &bias, &dst }, data_layout); + add_padding_x({ &weights }, data_layout, input_shape[0] % 4 == 0); // Don't add left padding if cl image will be used + // Create and configure function FunctionType conv; conv.configure(&src, &weights, &bias, &dst, info, act_info); @@ -171,8 +174,6 @@ protected: ARM_COMPUTE_ASSERT(bias.info()->is_resizable()); ARM_COMPUTE_ASSERT(dst.info()->is_resizable()); - add_padding_x({ &src, &bias, &dst }, data_layout); - // Allocate tensors src.allocator()->allocate(); weights.allocator()->allocate(); |