1 /*
2 * Copyright (c) 2017-2020 Arm Limited.
3 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
24 #include "arm_compute/runtime/CL/functions/CLHOGGradient.h"
25
26 #include "arm_compute/core/Helpers.h"
27 #include "arm_compute/core/Validate.h"
28 #include "arm_compute/runtime/CL/CLScheduler.h"
29 #include "src/core/CL/kernels/CLFillBorderKernel.h"
30 #include "src/core/CL/kernels/CLMagnitudePhaseKernel.h"
31 #include "support/MemorySupport.h"
32
33 using namespace arm_compute;
34
CLHOGGradient(std::shared_ptr<IMemoryManager> memory_manager)35 CLHOGGradient::CLHOGGradient(std::shared_ptr<IMemoryManager> memory_manager)
36 : _memory_group(std::move(memory_manager)),
37 _derivative(),
38 _mag_phase(support::cpp14::make_unique<CLMagnitudePhaseKernel>()),
39 _gx(),
40 _gy()
41 {
42 }
43
configure(ICLTensor * input,ICLTensor * output_magnitude,ICLTensor * output_phase,PhaseType phase_type,BorderMode border_mode,uint8_t constant_border_value)44 void CLHOGGradient::configure(ICLTensor *input, ICLTensor *output_magnitude, ICLTensor *output_phase, PhaseType phase_type, BorderMode border_mode, uint8_t constant_border_value)
45 {
46 configure(CLKernelLibrary::get().get_compile_context(), input, output_magnitude, output_phase, phase_type, border_mode, constant_border_value);
47 }
48
configure(const CLCompileContext & compile_context,ICLTensor * input,ICLTensor * output_magnitude,ICLTensor * output_phase,PhaseType phase_type,BorderMode border_mode,uint8_t constant_border_value)49 void CLHOGGradient::configure(const CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output_magnitude, ICLTensor *output_phase, PhaseType phase_type, BorderMode border_mode,
50 uint8_t constant_border_value)
51 {
52 ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::U8);
53 ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output_magnitude, 1, DataType::S16);
54 ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output_phase, 1, DataType::U8);
55
56 const TensorShape &shape_img = input->info()->tensor_shape();
57
58 // Allocate image memory
59 TensorInfo info(shape_img, Format::S16);
60 _gx.allocator()->init(info);
61 _gy.allocator()->init(info);
62
63 // Manage intermediate buffers
64 _memory_group.manage(&_gx);
65 _memory_group.manage(&_gy);
66
67 // Initialise derivate kernel
68 _derivative.configure(compile_context, input, &_gx, &_gy, border_mode, constant_border_value);
69
70 // Initialise magnitude/phase kernel
71 if(PhaseType::UNSIGNED == phase_type)
72 {
73 _mag_phase->configure(compile_context, &_gx, &_gy, output_magnitude, output_phase, MagnitudeType::L2NORM, PhaseType::UNSIGNED);
74 }
75 else
76 {
77 _mag_phase->configure(compile_context, &_gx, &_gy, output_magnitude, output_phase, MagnitudeType::L2NORM, PhaseType::SIGNED);
78 }
79
80 // Allocate intermediate tensors
81 _gx.allocator()->allocate();
82 _gy.allocator()->allocate();
83 }
84
run()85 void CLHOGGradient::run()
86 {
87 MemoryGroupResourceScope scope_mg(_memory_group);
88
89 // Run derivative
90 _derivative.run();
91
92 // Run magnitude/phase kernel
93 CLScheduler::get().enqueue(*_mag_phase);
94 }