diff options
author | John Richardson <john.richardson@arm.com> | 2018-02-05 15:12:22 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:37 +0000 |
commit | 7f4a8191a0fff69ec6c819e8d785a2c780388feb (patch) | |
tree | e027b6d011055f79d7de15b9b145aa621bf90411 /src/core/NEON/kernels/NEHOGDescriptorKernel.cpp | |
parent | c13021e335b3e395c9d1a3a9935baedb42aebf08 (diff) | |
download | ComputeLibrary-7f4a8191a0fff69ec6c819e8d785a2c780388feb.tar.gz |
COMPMID-597: Port HOGMultiDetection to new framework
Change-Id: I4b31b7f052a06bea4154d04c9926a0e076e28d73
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/126555
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: John Richardson <john.richardson@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEHOGDescriptorKernel.cpp')
-rw-r--r-- | src/core/NEON/kernels/NEHOGDescriptorKernel.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/NEHOGDescriptorKernel.cpp b/src/core/NEON/kernels/NEHOGDescriptorKernel.cpp index abe224e854..c204395586 100644 --- a/src/core/NEON/kernels/NEHOGDescriptorKernel.cpp +++ b/src/core/NEON/kernels/NEHOGDescriptorKernel.cpp @@ -786,7 +786,7 @@ void NEHOGBlockNormalizationKernel::run(const Window &window, const ThreadInfo & Window win_in(window); win_in.set_dimension_step(Window::DimX, _num_cells_per_block_stride.width); - win_in.set_dimension_step(Window::DimY, _num_cells_per_block_stride.height); + win_in.set(Window::DimY, Window::Dimension(0, 0, 0)); Iterator in(_input, win_in); Iterator out(_output, window); @@ -794,7 +794,7 @@ void NEHOGBlockNormalizationKernel::run(const Window &window, const ThreadInfo & // Normalises blocks execute_window_loop(window, [&](const Coordinates & id) { - const auto input_row_ptr = reinterpret_cast<const float *>(in.ptr()); + const auto input_row_ptr = reinterpret_cast<const float *>(in.ptr() + id.y() * _num_cells_per_block_stride.height * _input->info()->strides_in_bytes()[Window::DimY]); const auto out_row_ptr = reinterpret_cast<float *>(out.ptr()); // Execute normalization function |