diff options
author | Anthony Barbier <anthony.barbier@arm.com> | 2017-09-04 18:44:23 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 13:03:09 +0100 |
commit | 6ff3b19ee6120edf015fad8caab2991faa3070af (patch) | |
tree | a7a6dcd16dfd56d79fa1b56a313caeebcc939b68 /src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp | |
download | ComputeLibrary-6ff3b19ee6120edf015fad8caab2991faa3070af.tar.gz |
COMPMID-344 Updated doxygen
Change-Id: I32f7b84daa560e460b77216add529c8fa8b327ae
Diffstat (limited to 'src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp | 371 |
1 files changed, 371 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp b/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp new file mode 100644 index 0000000000..d3e62b069e --- /dev/null +++ b/src/core/NEON/kernels/NEArithmeticSubtractionKernel.cpp @@ -0,0 +1,371 @@ +/* + * Copyright (c) 2016, 2017 ARM Limited. + * + * SPDX-License-Identifier: MIT + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to + * deal in the Software without restriction, including without limitation the + * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or + * sell copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in all + * copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE + * SOFTWARE. + */ +#include "arm_compute/core/NEON/kernels/NEArithmeticSubtractionKernel.h" + +#include "arm_compute/core/Error.h" +#include "arm_compute/core/Helpers.h" +#include "arm_compute/core/ITensor.h" +#include "arm_compute/core/TensorInfo.h" +#include "arm_compute/core/Validate.h" + +#include <algorithm> +#include <arm_neon.h> +#include <cstdint> +#include <map> +#include <string> + +using namespace arm_compute; + +namespace arm_compute +{ +class Coordinates; +} // namespace arm_compute + +namespace +{ +void sub_wrap_U8_U8_U8(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t ta1 = vld1q_u8(input1.ptr()); + const uint8x16_t ta2 = vld1q_u8(input2.ptr()); + + vst1q_u8(output.ptr(), vsubq_u8(ta1, ta2)); + }, + input1, input2, output); +} + +void sub_saturate_U8_U8_U8(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t ta1 = vld1q_u8(input1.ptr()); + const uint8x16_t ta2 = vld1q_u8(input2.ptr()); + + vst1q_u8(output.ptr(), vqsubq_u8(ta1, ta2)); + }, + input1, input2, output); +} + +void sub_wrap_S16_S16_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const int16x8x2_t ta1 = vld2q_s16(reinterpret_cast<const int16_t *>(input1.ptr())); + const int16x8x2_t ta2 = vld2q_s16(reinterpret_cast<const int16_t *>(input2.ptr())); + + const int16x8x2_t ta3 = + { + { + vsubq_s16(ta1.val[0], ta2.val[0]), + vsubq_s16(ta1.val[1], ta2.val[1]) + } + }; + + vst2q_s16(reinterpret_cast<int16_t *>(output.ptr()), ta3); + }, + input1, input2, output); +} + +void sub_saturate_S16_S16_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const int16x8x2_t ta1 = vld2q_s16(reinterpret_cast<const int16_t *>(input1.ptr())); + const int16x8x2_t ta2 = vld2q_s16(reinterpret_cast<const int16_t *>(input2.ptr())); + + const int16x8x2_t ta3 = + { + { + vqsubq_s16(ta1.val[0], ta2.val[0]), + vqsubq_s16(ta1.val[1], ta2.val[1]) + } + }; + + vst2q_s16(reinterpret_cast<int16_t *>(output.ptr()), ta3); + }, + input1, input2, output); +} + +void sub_F32_F32_F32(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const float32x4x4_t ta1 = vld4q_f32(reinterpret_cast<const float *>(input1.ptr())); + const float32x4x4_t ta2 = vld4q_f32(reinterpret_cast<const float *>(input2.ptr())); + + const float32x4x4_t ta3 = + { + { + vsubq_f32(ta1.val[0], ta2.val[0]), + vsubq_f32(ta1.val[1], ta2.val[1]), + vsubq_f32(ta1.val[2], ta2.val[2]), + vsubq_f32(ta1.val[3], ta2.val[3]), + } + }; + + vst4q_f32(reinterpret_cast<float *>(output.ptr()), ta3); + }, + input1, input2, output); +} +void sub_wrap_S16_U8_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t bv_0 = vld1q_u8(input2.ptr()); + int16x8_t a1_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input1.ptr())); + int16x8_t a2_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input1.ptr()) + 8); + + a1_0 = vsubq_s16(a1_0, vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0)))); + a2_0 = vsubq_s16(a2_0, vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0)))); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} + +void sub_saturate_S16_U8_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t bv_0 = vld1q_u8(input2.ptr()); + int16x8_t a1_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input1.ptr())); + int16x8_t a2_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input1.ptr()) + 8); + + a1_0 = vqsubq_s16(a1_0, vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0)))); + a2_0 = vqsubq_s16(a2_0, vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0)))); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} + +void sub_wrap_U8_S16_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t bv_0 = vld1q_u8(input1.ptr()); + int16x8_t a1_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input2.ptr())); + int16x8_t a2_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input2.ptr()) + 8); + + a1_0 = vsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0))), a1_0); + a2_0 = vsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0))), a2_0); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} + +void sub_saturate_U8_S16_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t bv_0 = vld1q_u8(input1.ptr()); + int16x8_t a1_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input2.ptr())); + int16x8_t a2_0 = vld1q_s16(reinterpret_cast<const int16_t *>(input2.ptr()) + 8); + + a1_0 = vqsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0))), a1_0); + a2_0 = vqsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0))), a2_0); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} + +void sub_wrap_U8_U8_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t av_0 = vld1q_u8(input1.ptr()); + const uint8x16_t bv_0 = vld1q_u8(input2.ptr()); + + const int16x8_t a1_0 = vsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(av_0))), + vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0)))); + const int16x8_t a2_0 = vsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(av_0))), + vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0)))); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} + +void sub_saturate_U8_U8_S16(const ITensor *in1, const ITensor *in2, ITensor *out, const Window &window) +{ + Iterator input1(in1, window); + Iterator input2(in2, window); + Iterator output(out, window); + + execute_window_loop(window, [&](const Coordinates & id) + { + const uint8x16_t av_0 = vld1q_u8(input1.ptr()); + const uint8x16_t bv_0 = vld1q_u8(input2.ptr()); + + const int16x8_t a1_0 = vqsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(av_0))), + vreinterpretq_s16_u16(vmovl_u8(vget_low_u8(bv_0)))); + const int16x8_t a2_0 = vqsubq_s16(vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(av_0))), + vreinterpretq_s16_u16(vmovl_u8(vget_high_u8(bv_0)))); + + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()), a1_0); + vst1q_s16(reinterpret_cast<int16_t *>(output.ptr()) + 8, a2_0); + }, + input1, input2, output); +} +} // namespace + +NEArithmeticSubtractionKernel::NEArithmeticSubtractionKernel() + : _func(nullptr), _input1(nullptr), _input2(nullptr), _output(nullptr) +{ +} + +void NEArithmeticSubtractionKernel::configure(const ITensor *input1, const ITensor *input2, ITensor *output, ConvertPolicy policy) +{ + ARM_COMPUTE_ERROR_ON_NULLPTR(input1, input2, output); + + set_shape_if_empty(*output->info(), input1->info()->tensor_shape()); + + if(input1->info()->data_type() == DataType::S16 || input2->info()->data_type() == DataType::S16) + { + set_format_if_unknown(*output->info(), Format::S16); + } + else if(input1->info()->data_type() == DataType::F32 || input2->info()->data_type() == DataType::F32) + { + set_format_if_unknown(*output->info(), Format::F32); + } + + ARM_COMPUTE_ERROR_ON_MISMATCHING_SHAPES(input1, input2, output); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input1, 1, DataType::U8, DataType::S16, DataType::F32); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input2, 1, DataType::U8, DataType::S16, DataType::F32); + ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output, 1, DataType::U8, DataType::S16, DataType::F32); + ARM_COMPUTE_ERROR_ON_MSG(output->info()->data_type() == DataType::U8 && (input1->info()->data_type() != DataType::U8 || input2->info()->data_type() != DataType::U8), + "Output can only be U8 if both inputs are U8"); + + static std::map<std::string, SubFunction *> map_function = + { + { "sub_wrap_U8_U8_U8", &sub_wrap_U8_U8_U8 }, + { "sub_wrap_U8_U8_S16", &sub_wrap_U8_U8_S16 }, + { "sub_saturate_U8_U8_U8", &sub_saturate_U8_U8_U8 }, + { "sub_saturate_U8_U8_S16", &sub_saturate_U8_U8_S16 }, + { "sub_wrap_U8_S16_S16", &sub_wrap_U8_S16_S16 }, + { "sub_wrap_S16_U8_S16", &sub_wrap_S16_U8_S16 }, + { "sub_saturate_U8_S16_S16", &sub_saturate_U8_S16_S16 }, + { "sub_saturate_S16_U8_S16", &sub_saturate_S16_U8_S16 }, + { "sub_wrap_S16_S16_S16", &sub_wrap_S16_S16_S16 }, + { "sub_saturate_S16_S16_S16", &sub_saturate_S16_S16_S16 }, + { "sub_wrap_F32_F32_F32", &sub_F32_F32_F32 }, + { "sub_saturate_F32_F32_F32", &sub_F32_F32_F32 }, + }; + + _input1 = input1; + _input2 = input2; + _output = output; + + std::string function_to_call("sub_"); + function_to_call += policy == ConvertPolicy::WRAP ? "wrap_" : "saturate_"; + function_to_call += string_from_data_type(input1->info()->data_type()) + "_"; + function_to_call += string_from_data_type(input2->info()->data_type()) + "_"; + function_to_call += string_from_data_type(output->info()->data_type()); + + auto it = map_function.find(function_to_call); + + if(it != map_function.end()) + { + _func = it->second; + } + else + { + ARM_COMPUTE_ERROR("You called subtract with the wrong image formats"); + } + + constexpr unsigned int num_elems_processed_per_iteration = 16; + + // Configure kernel window + Window win = calculate_max_window(*input1->info(), Steps(num_elems_processed_per_iteration)); + AccessWindowHorizontal output_access(output->info(), 0, num_elems_processed_per_iteration); + + update_window_and_padding(win, + AccessWindowHorizontal(input1->info(), 0, num_elems_processed_per_iteration), + AccessWindowHorizontal(input2->info(), 0, num_elems_processed_per_iteration), + output_access); + + ValidRegion valid_region = intersect_valid_regions(input1->info()->valid_region(), + input2->info()->valid_region()); + + output_access.set_valid_region(win, valid_region); + + INEKernel::configure(win); +} + +void NEArithmeticSubtractionKernel::run(const Window &window) +{ + ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); + ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(INEKernel::window(), window); + ARM_COMPUTE_ERROR_ON(_func == nullptr); + + (*_func)(_input1, _input2, _output, window); +} |