diff options
Diffstat (limited to 'src/core/CL/cl_kernels/nhwc/batch_to_space.cl')
-rw-r--r-- | src/core/CL/cl_kernels/nhwc/batch_to_space.cl | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/core/CL/cl_kernels/nhwc/batch_to_space.cl b/src/core/CL/cl_kernels/nhwc/batch_to_space.cl index 16141bcd2e..b910a753a6 100644 --- a/src/core/CL/cl_kernels/nhwc/batch_to_space.cl +++ b/src/core/CL/cl_kernels/nhwc/batch_to_space.cl @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2021, 2023 Arm Limited. + * Copyright (c) 2018-2021, 2023-2024 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -63,7 +63,7 @@ __kernel void batch_to_space_nhwc( TENSOR3D_DECLARATION(output)) { Tensor3D out = CONVERT_TO_TENSOR3D_STRUCT(output); - Tensor4D in = CONVERT_TO_TENSOR4D_STRUCT_NO_STEP(input, 0); + Tensor4D in = CONVERT_TO_TENSOR4D_STRUCT_NO_STEP(input); Vector block = CONVERT_TO_VECTOR_STRUCT_NO_STEP(block_shape); const int block_x = *((__global int *)vector_offset(&block, 0)); @@ -113,7 +113,7 @@ __kernel void batch_to_space_static_nhwc( TENSOR3D_DECLARATION(output)) { Tensor3D out = CONVERT_TO_TENSOR3D_STRUCT(output); - Tensor4D in = CONVERT_TO_TENSOR4D_STRUCT_NO_STEP(input, 0); + Tensor4D in = CONVERT_TO_TENSOR4D_STRUCT_NO_STEP(input); const int block_x = BLOCK_SHAPE_X; const int block_y = BLOCK_SHAPE_Y; @@ -128,4 +128,4 @@ __kernel void batch_to_space_static_nhwc( *((__global DATA_TYPE *)out.ptr) = *((__global DATA_TYPE *)tensor4D_offset(&in, z, in_x, in_y, in_batch)); } -#endif // defined(DATA_TYPE) && defined(BATCH_SIZE) && defined(BLOCK_SHAPE_X) && defined(BLOCK_SHAPE_Y)
\ No newline at end of file +#endif // defined(DATA_TYPE) && defined(BATCH_SIZE) && defined(BLOCK_SHAPE_X) && defined(BLOCK_SHAPE_Y) |