diff options
Diffstat (limited to 'src/core/NEON')
-rw-r--r-- | src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp | 4 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEIm2ColKernel.cpp | 6 |
2 files changed, 5 insertions, 5 deletions
diff --git a/src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp b/src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp index 618a1baf07..d012cbfded 100644 --- a/src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp +++ b/src/core/NEON/kernels/NEDepthToSpaceLayerKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 Arm Limited. + * Copyright (c) 2019-2020 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -71,7 +71,7 @@ NEDepthToSpaceLayerKernel::NEDepthToSpaceLayerKernel() void NEDepthToSpaceLayerKernel::configure(const ITensor *input, ITensor *output, int32_t block_shape) { ARM_COMPUTE_ERROR_ON_NULLPTR(input, output); - TensorShape output_shape = compute_depth_to_space_shape(input->info(), block_shape); + TensorShape output_shape = compute_depth_to_space_shape(input->info()->tensor_shape(), input->info()->data_layout(), block_shape); // Output auto inizialitation if not yet initialized auto_init_if_empty(*output->info(), input->info()->clone()->set_tensor_shape(output_shape)); diff --git a/src/core/NEON/kernels/NEIm2ColKernel.cpp b/src/core/NEON/kernels/NEIm2ColKernel.cpp index 1a2b95e8d6..6eae0541aa 100644 --- a/src/core/NEON/kernels/NEIm2ColKernel.cpp +++ b/src/core/NEON/kernels/NEIm2ColKernel.cpp @@ -161,7 +161,7 @@ inline void linearize_volume_nchw(const uint8_t *const in_ptr, if((y < 0 || y >= input_h) && has_pads) { // All the values will be the offset (will be zeros when not quantized) - memset(out_ptr, pad_value, kernel_width * sizeof(T)); + memset(static_cast<void *>(out_ptr), pad_value, kernel_width * sizeof(T)); out_ptr += kernel_width; } else @@ -224,7 +224,7 @@ inline void linearize_volume_nhwc(const uint8_t *const in_ptr, { if(y < 0 || y >= input_h) { - memset(out_ptr, pad_value, pad_quant * element_size); + memset(static_cast<void *>(out_ptr), pad_value, pad_quant * element_size); out_ptr += pad_quant; } else if(dilation_x > 1 || start_x < 0 || end_x >= input_w || input_stride_y != input_c * element_size) @@ -233,7 +233,7 @@ inline void linearize_volume_nhwc(const uint8_t *const in_ptr, { if(x < 0 || x >= input_w) { - memset(out_ptr, pad_value, input_c * element_size); + memset(static_cast<void *>(out_ptr), pad_value, input_c * element_size); out_ptr += input_c; } else |