diff options
Diffstat (limited to 'src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp b/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp index 4da3e245c0..70af541ef1 100644 --- a/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp +++ b/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp @@ -149,6 +149,9 @@ void CLHeightConcatenateLayerKernel::configure(const ICLTensor *input, unsigned ARM_COMPUTE_ERROR_THROW_ON(std::get<0>(win_config)); ICLKernel::configure_internal(std::get<1>(win_config)); + + // Set output valid region + output->info()->set_valid_region(ValidRegion(Coordinates(), output->info()->tensor_shape())); } void CLHeightConcatenateLayerKernel::run(const Window &window, cl::CommandQueue &queue) |