| 136 | } |
| 137 | |
| 138 | void ClQuantizeKernel::run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) |
| 139 | { |
| 140 | ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this); |
| 141 | ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); |
| 142 | |
| 143 | auto src = utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::ACL_SRC)); |
| 144 | auto dst = utils::cast::polymorphic_downcast<ICLTensor *>(tensors.get_tensor(TensorType::ACL_DST)); |
| 145 | |
| 146 | Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), 3); |
| 147 | Window slice = window_collapsed.first_slice_window_3D(); |
| 148 | |
| 149 | do |
| 150 | { |
| 151 | unsigned int idx = 0; |
| 152 | add_3D_tensor_argument(idx, src, slice); |
| 153 | add_3D_tensor_argument(idx, dst, slice); |
| 154 | enqueue(queue, *this, slice, lws_hint()); |
| 155 | } while (window_collapsed.slide_window_slice_3D(slice)); |
| 156 | } |
| 157 | } // namespace kernels |
| 158 | } // namespace opencl |
| 159 | } // namespace arm_compute |
nothing calls this directly
no test coverage detected