diff options
-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 { |