| 152 | } |
| 153 | |
| 154 | void CLReverseKernel::run(const Window &window, cl::CommandQueue &queue) |
| 155 | { |
| 156 | ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); |
| 157 | ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); |
| 158 | |
| 159 | Window collapsed = window.collapse(ICLKernel::window(), Window::DimZ); |
| 160 | Window slice = collapsed.first_slice_window_4D(); |
| 161 | Window axis_slice = collapsed.first_slice_window_1D(); |
| 162 | |
| 163 | do |
| 164 | { |
| 165 | unsigned int idx = 0; |
| 166 | add_4D_tensor_argument(idx, _input, slice); |
| 167 | add_1D_tensor_argument(idx, _axis, axis_slice); |
| 168 | add_4D_tensor_argument(idx, _output, slice); |
| 169 | enqueue(queue, *this, slice, lws_hint()); |
| 170 | } while (collapsed.slide_window_slice_4D(slice)); |
| 171 | } |
| 172 | } // namespace arm_compute |
nothing calls this directly
no test coverage detected