MCPcopy Create free account
hub / github.com/ARM-software/ComputeLibrary / run_op

Method run_op

src/core/CL/kernels/CLStridedSliceKernel.cpp:192–211  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

190}
191
192void CLStridedSliceKernel::run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue)
193{
194 ARM_COMPUTE_ERROR_ON_UNCONFIGURED_KERNEL(this);
195 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window);
196
197 const auto src =
198 utils::cast::polymorphic_downcast<const ICLTensor *>(tensors.get_const_tensor(TensorType::ACL_SRC));
199 auto dst = utils::cast::polymorphic_downcast<ICLTensor *>(tensors.get_tensor(TensorType::ACL_DST));
200
201 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ);
202 Window slice = window_collapsed.first_slice_window_4D();
203
204 do
205 {
206 unsigned int idx = 0;
207 add_4D_tensor_argument(idx, src, slice);
208 add_4D_tensor_argument(idx, dst, slice);
209 enqueue(queue, *this, slice, lws_hint());
210 } while (window_collapsed.slide_window_slice_4D(slice));
211}
212} // namespace arm_compute

Callers

nothing calls this directly

Calls 5

get_const_tensorMethod · 0.80
collapse_if_possibleMethod · 0.80
first_slice_window_4DMethod · 0.80
slide_window_slice_4DMethod · 0.80
get_tensorMethod · 0.45

Tested by

no test coverage detected