diff options
| author | Giorgio Arena <giorgio.arena@arm.com> | 2021-02-17 16:04:05 +0000 |
|---|---|---|
| committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-02-18 14:01:07 +0000 |
| commit | 9a09180c1cf44b6db52d2166df1ad92abbe1dbae (patch) | |
| tree | d0ecca5926069ce8f98d02777bb5c035e32eaf8e /src/core/CL/kernels/CLDirectConvolutionLayerKernel.h | |
| parent | d820ed780c808ce8862b91da315f50b9531d2a8b (diff) | |
| download | ComputeLibrary-9a09180c1cf44b6db52d2166df1ad92abbe1dbae.tar.gz | |
Set CLDirectConvolutionLayerKernel NCHW _border_size to input padding
Change-Id: I5802c470683647b7426b3b6e7d17280cabc32163
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5100
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLDirectConvolutionLayerKernel.h')
| -rw-r--r-- | src/core/CL/kernels/CLDirectConvolutionLayerKernel.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.h b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.h index 5cd674f631..0257d0c2dd 100644 --- a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.h +++ b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -120,6 +120,7 @@ public: BorderSize _border_size; int _conv_stride_x; int _conv_stride_y; + PadStrideInfo _conv_info; }; } // namespace arm_compute #endif /*ARM_COMPUTE_CLDIRECTCONVOLUTIONLAYERKERNEL_H */ |