Home
last modified time | relevance | path

Searched refs:ICLKernel (Results 1 – 25 of 150) sorted by relevance

123456

/external/ComputeLibrary/src/core/CL/
DICLKernel.cpp32 void arm_compute::enqueue(cl::CommandQueue &queue, ICLKernel &kernel, const Window &window, const c… in enqueue()
46 cl::NDRange gws = ICLKernel::gws_from_window(window); in enqueue()
88 void ICLKernel::add_tensor_argument(unsigned &idx, const ICLTensor *tensor, const Window &window) in add_tensor_argument()
119 void ICLKernel::add_3d_tensor_nhw_argument(unsigned int &idx, const ICLTensor *tensor) in add_3d_tensor_nhw_argument()
144 void ICLKernel::add_4d_tensor_nhwc_argument(unsigned int &idx, const ICLTensor *tensor) in add_4d_tensor_nhwc_argument()
172 template void ICLKernel::add_tensor_argument<1>(unsigned &idx, const ICLTensor *tensor, const Windo…
173 template void ICLKernel::add_tensor_argument<2>(unsigned &idx, const ICLTensor *tensor, const Windo…
174 template void ICLKernel::add_tensor_argument<3>(unsigned &idx, const ICLTensor *tensor, const Windo…
175 template void ICLKernel::add_tensor_argument<4>(unsigned &idx, const ICLTensor *tensor, const Windo…
176 template void ICLKernel::add_tensor_argument<5>(unsigned &idx, const ICLTensor *tensor, const Windo…
[all …]
DICLKernel.h67 class ICLKernel : public IKernel
126 ICLKernel() in ICLKernel() function
485 void enqueue(cl::CommandQueue &queue, ICLKernel &kernel, const Window &window, const cl::NDRange &l…
496 void ICLKernel::add_array_argument(unsigned &idx, const ICLArray<T> *array, const Strides &strides,… in add_array_argument()
/external/ComputeLibrary/arm_compute/runtime/CL/
DCLTuner.h37 class ICLKernel; variable
124 void tune_kernel_static(ICLKernel &kernel) override;
125 void tune_kernel_dynamic(ICLKernel &kernel) override;
126 void tune_kernel_dynamic(ICLKernel &kernel, ITensorPack &tensors) override;
144 void do_tune_kernel_dynamic(ICLKernel &kernel, IKernelData *data);
152 CLTuningParams find_optimal_tuning_params(ICLKernel &kernel, IKernelData *data);
DICLTuner.h31 class ICLKernel; variable
46 virtual void tune_kernel_static(ICLKernel &kernel) = 0;
53 virtual void tune_kernel_dynamic(ICLKernel &kernel) = 0;
59 virtual void tune_kernel_dynamic(ICLKernel &kernel, ITensorPack &tensors) = 0;
DCLScheduler.h40 class ICLKernel; variable
90 void enqueue(ICLKernel &kernel, bool flush = true);
97 void enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush = true);
172 void tune_kernel_static(ICLKernel &kernel);
183 void enqueue_common(ICLKernel &kernel, ITensorPack &tensors, bool flush);
DCLHelpers.h35 class ICLKernel; variable
54 void schedule_kernel_on_ctx(CLRuntimeContext *ctx, ICLKernel *kernel, bool flush = true);
DICLSimpleFunction.h36 class ICLKernel; variable
63 std::unique_ptr<ICLKernel> _kernel; /**< Kernel to run */
DICLOperator.h37 class ICLKernel; variable
64 std::unique_ptr<ICLKernel> _kernel;
/external/ComputeLibrary/src/core/CL/kernels/
DCLFillBorderKernel.cpp40 : ICLKernel(), _tensor(nullptr) in CLFillBorderKernel()
55 ICLKernel::add_argument<T>(idx, static_cast<T>(value)); in set_constant_border()
113 ICLKernel::add_argument<cl_uint>(idx, valid_width); in configure()
114 ICLKernel::add_argument<cl_uint>(idx, valid_height); in configure()
115 ICLKernel::add_argument<cl_int2>(idx, valid_region_coords); in configure()
159 ICLKernel::configure_internal(win); in configure()
185 ARM_COMPUTE_ERROR_ON_MISMATCHING_WINDOWS(ICLKernel::window(), window); in run_op()
187 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op()
208 ARM_COMPUTE_ERROR_ON_MISMATCHING_WINDOWS(ICLKernel::window(), window); in run()
210 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run()
DCLFFTScaleKernel.cpp98 ICLKernel::configure_internal(win); in configure()
122 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run()
124 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run()
DCLReverseKernel.cpp103 ICLKernel::configure_internal(win); in configure()
126 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run()
128 Window collapsed = window.collapse(ICLKernel::window(), Window::DimZ); in run()
DCLStackLayerKernel.cpp109 ICLKernel::configure_internal(win_config.second); in configure()
125 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run()
130 Window collapsed = window.collapse(ICLKernel::window(), Window::DimZ); in run()
DCLTileKernel.cpp114 ICLKernel::configure_internal(win); in configure()
138 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run()
140 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run()
DCLFFTDigitReverseKernel.cpp102 ICLKernel::configure_internal(win_config.second); in configure()
126 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run()
128 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run()
/external/ComputeLibrary/src/runtime/CL/
DCLTuner.cpp46 virtual void do_run(ICLKernel &kernel, cl::CommandQueue &queue) = 0;
55 void do_run(ICLKernel &kernel, cl::CommandQueue &queue) override in do_run()
88 void CLTuner::tune_kernel_static(ICLKernel &kernel) in tune_kernel_static()
93 void CLTuner::tune_kernel_dynamic(ICLKernel &kernel) in tune_kernel_dynamic()
99 void CLTuner::do_tune_kernel_dynamic(ICLKernel &kernel, IKernelData *data) in do_tune_kernel_dynamic()
138 void CLTuner::tune_kernel_dynamic(ICLKernel &kernel, ITensorPack &tensors) in tune_kernel_dynamic()
150 CLTuningParams CLTuner::find_optimal_tuning_params(ICLKernel &kernel, IKernelData *data) in find_optimal_tuning_params()
202 cl::NDRange gws = ICLKernel::gws_from_window(kernel.window()); in find_optimal_tuning_params()
DCLScheduler.cpp82 void CLScheduler::tune_kernel_static(ICLKernel &kernel) in tune_kernel_static()
165 void CLScheduler::enqueue_common(ICLKernel &kernel, ITensorPack &tensors, bool flush) in enqueue_common()
205 void CLScheduler::enqueue(ICLKernel &kernel, bool flush) in enqueue()
211 void CLScheduler::enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush) in enqueue_op()
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLInstanceNormalizationLayer.h38 class ICLKernel; variable
115 std::unique_ptr<ICLKernel> _inst_norm_kernel; /**< Kernel to run */
116 std::unique_ptr<ICLKernel> _mean_var_kernel; /**< Kernel to run */
/external/ComputeLibrary/src/gpu/cl/kernels/
DClDequantizeKernel.cpp116 ICLKernel::configure_internal(win); in configure()
130 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
138 …ed_per_channel ? window.collapse_if_possible(ICLKernel::window(), 4) : window.collapse_if_possible… in run_op()
DClCastKernel.cpp115 ICLKernel::configure_internal(win); in configure()
120 ICLKernel::configure_internal(collapsed_window); in configure()
147 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
154 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op()
DClCopyKernel.cpp124 ICLKernel::configure_internal(win_config); in configure()
139 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
161 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op()
DClElementwiseUnaryKernel.cpp132 ICLKernel::configure_internal(win); in configure()
149 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
151 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op()
DClPermuteKernel.cpp112 ICLKernel::configure_internal(win); in configure()
127 ARM_COMPUTE_ERROR_ON_MISMATCHING_WINDOWS(ICLKernel::window(), window); in run_op()
132 Window slice_in = window.first_slice_window_4D().collapse(ICLKernel::window(), 2, 4); in run_op()
DClQuantizeKernel.cpp147 ICLKernel::configure_internal(win); in configure()
161 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
166 Window window_collapsed = window.collapse_if_possible(ICLKernel::window(), 3); in run_op()
DClGemmLowpQuantizeDownInt32ScaleByFloatKernel.cpp126 ICLKernel::configure_internal(win); in configure()
134 ARM_COMPUTE_ERROR_ON_INVALID_SUBWINDOW(ICLKernel::window(), window); in run_op()
141 Window collapsed = window.collapse_if_possible(ICLKernel::window(), Window::DimZ); in run_op()
/external/ComputeLibrary/tests/CL/
DHelper.h121 std::unique_ptr<ICLKernel> _kernel{}; /**< Kernel to run */
219 std::unique_ptr<ICLKernel> _kernel{}; /**< Kernel to run */
251 std::unique_ptr<ICLKernel> _border_handler{ nullptr }; /**< Kernel to handle borders */
252 std::unique_ptr<ICLKernel> _kernel{}; /**< Kernel to run */

123456