Home
last modified time | relevance | path

Searched refs:window_collapsed (Results 1 – 12 of 12) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/kernels/
DClReshapeKernel.cpp119 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
120 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
127 add_3D_tensor_argument(idx, src, window_collapsed); in run_op()
128 add_3D_tensor_argument(idx, dst, window_collapsed); in run_op()
DClSoftmaxKernel.cpp238 Window window_collapsed = window.collapse_if_possible(IClKernel::window(), Window::DimZ); in run_op() local
245 window_collapsed.set(Window::DimX, Window::Dimension(0, _grid_size, 1)); in run_op()
249 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
260 while(window_collapsed.slide_window_slice_3D(slice)); in run_op()
330 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
331 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
345 while(window_collapsed.slide_window_slice_3D(slice)); in run_op()
DClQuantizeKernel.cpp166 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), 3); in run_op() local
167 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
176 while(window_collapsed.slide_window_slice_3D(slice)); in run_op()
DClPool2dKernel.cpp310 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
316 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
329 while(window_collapsed.slide_window_slice_3D(slice)); in run_op()
336 Window slice = window_collapsed.first_slice_window_4D(); in run_op()
337 Window in_slice = window_collapsed.first_slice_window_4D(); in run_op()
DClWinogradInputTransformKernel.cpp239 Window window_collapsed = window.collapse_if_possible(IClKernel::window(), Window::DimZ); in run_op() local
243 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
258 Window slice = window_collapsed.first_slice_window_3D(); in run_op()
277 while(window_collapsed.slide_window_slice_3D(slice)); in run_op()
DClIm2ColKernel.cpp374 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
375 window_collapsed.set_dimension_step(Window::DimZ, 1); in run_op()
384 const Window first_slice_3d = window_collapsed.first_slice_window_3D(); in run_op()
433 …while(window_collapsed.slide_window_slice_3D(slice) && window_output.slide_window_slice_2D(slice_o… in run_op()
DClPool3dKernel.cpp245 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
252 enqueue(queue, *this, window_collapsed, lws_hint()); in run_op()
DClWinogradOutputTransformKernel.cpp274 Window window_collapsed = window.collapse_if_possible(IClKernel::window(), Window::DimZ); in run_op() local
277 Window slice = window_collapsed.first_slice_window_4D(); in run_op()
/external/ComputeLibrary/src/core/CL/kernels/
DCLGatherKernel.cpp126 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run() local
128 add_4D_tensor_argument(idx, _input, window_collapsed); in run()
129 add_1D_tensor_argument(idx, _indices, window_collapsed); in run()
130 add_4D_tensor_argument(idx, _output, window_collapsed); in run()
131 enqueue(queue, *this, window_collapsed, lws_hint()); in run()
DCLStridedSliceKernel.cpp175 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op() local
176 Window slice = window_collapsed.first_slice_window_4D(); in run_op()
185 while(window_collapsed.slide_window_slice_4D(slice)); in run_op()
DCLNormalizationLayerKernel.cpp243 …Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), collapsed_dimensi… in run() local
244 Window slice = window_collapsed.first_slice_window_3D(); in run()
253 while(window_collapsed.slide_window_slice_3D(slice)); in run()
DCLDepthwiseConvolutionLayerNativeKernel.cpp364 Window window_collapsed = window.collapse(ICLKernel::window(), Window::DimZ); in run() local
366 Window slice = window_collapsed.first_slice_window_4D(); in run()