diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2021-03-19 15:06:07 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-03-29 19:48:22 +0000 |
commit | 702dc0c71f2b2830b63e3b4079ede0ef76377f0a (patch) | |
tree | afb0dacd74a8650350802aea8043822503c2f8b2 /src/core/CL/kernels/CLIm2ColKernel.cpp | |
parent | 26c5c0fc10cbee1b523af0b27f884719dc50820a (diff) | |
download | ComputeLibrary-702dc0c71f2b2830b63e3b4079ede0ef76377f0a.tar.gz |
Remove usage of valid window region CL - NHWC
Resolves: COMPMID-4153
Change-Id: Ib0d60c9acaac8aaf3946c62fc2d740b5ec6cee5c
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5301
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLIm2ColKernel.cpp')
-rw-r--r-- | src/core/CL/kernels/CLIm2ColKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/CL/kernels/CLIm2ColKernel.cpp b/src/core/CL/kernels/CLIm2ColKernel.cpp index 07309de83c..671058106e 100644 --- a/src/core/CL/kernels/CLIm2ColKernel.cpp +++ b/src/core/CL/kernels/CLIm2ColKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017-2020 Arm Limited. + * Copyright (c) 2017-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -147,7 +147,7 @@ std::pair<Status, Window> validate_and_configure_window(ITensorInfo *input, ITen win = calculate_max_window(*input, Steps()); } } - output->set_valid_region(ValidRegion(Coordinates(), output->tensor_shape())); + // set the Z dimension's step same size as the whole dimension so that one can't split across the Z dimension win.set_dimension_step(Window::DimZ, win[Window::DimZ].end() - win[Window::DimZ].start()); |