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:18 +0000 |
| commit | 95235f49c2389a8013c4ca68affe29745b257011 (patch) | |
| tree | 347fa0e4868106cbd85447318a3c61326fd10ec2 /src/core/CL/kernels/CLDirectConvolutionLayerKernel.h | |
| parent | 7b32098ff6933f0421fab659c25620d47c33fc8a (diff) | |
| download | ComputeLibrary-95235f49c2389a8013c4ca68affe29745b257011.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 */ |