diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-03-15 18:02:21 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | 6528aa20e768f2d801328aa164d672b7fdfe266f (patch) | |
tree | f47ac1e3ec04a146ac78837d24e20f67d9c4b898 /src/core | |
parent | d2fab7315bac3a586f2f1b1c8d64f2441f89ca64 (diff) | |
download | ComputeLibrary-6528aa20e768f2d801328aa164d672b7fdfe266f.tar.gz |
COMPMID-959: Set output valid region to CLPermuteKernel.
Change-Id: Ieb62347c71cef5ac8941e632c8afa678f5db6f21
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124813
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core')
-rw-r--r-- | src/core/CL/kernels/CLPermuteKernel.cpp | 5 | ||||
-rw-r--r-- | src/core/SubTensorInfo.cpp | 1 |
2 files changed, 6 insertions, 0 deletions
diff --git a/src/core/CL/kernels/CLPermuteKernel.cpp b/src/core/CL/kernels/CLPermuteKernel.cpp index da34448c02..02fc667aee 100644 --- a/src/core/CL/kernels/CLPermuteKernel.cpp +++ b/src/core/CL/kernels/CLPermuteKernel.cpp @@ -114,6 +114,11 @@ void CLPermuteKernel::configure(const ICLTensor *input, ICLTensor *output, const // Configure kernel window Window win = calculate_max_window(*input->info(), Steps()); + // The CLPermute doesn't need padding so update_window_and_padding() can be skipped + Coordinates coord; + coord.set_num_dimensions(output->info()->num_dimensions()); + output->info()->set_valid_region(ValidRegion(coord, output->info()->tensor_shape())); + ICLKernel::configure(win); } diff --git a/src/core/SubTensorInfo.cpp b/src/core/SubTensorInfo.cpp index 836c3794c2..237f1333f2 100644 --- a/src/core/SubTensorInfo.cpp +++ b/src/core/SubTensorInfo.cpp @@ -100,6 +100,7 @@ ITensorInfo &SubTensorInfo::set_tensor_shape(const TensorShape &shape) if(_parent->tensor_shape().total_size() != 0 && !_extend_parent) { ARM_COMPUTE_ERROR_ON_INVALID_SUBTENSOR(_parent->tensor_shape(), _coords, shape); + _valid_region = ValidRegion{ _coords, shape }; } else if(_extend_parent) // Extend parent shape, configure if specified { |