Searched refs:_col2im_kernel (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClGemmConv2d.cpp | 56 …, _im2col_kernel(nullptr), _mm_gemm(nullptr), _mm_gemmlowp(nullptr), _col2im_kernel(nullptr), _act… in ClGemmConv2d()
327 _col2im_kernel = std::make_unique<opencl::kernels::ClCol2ImKernel>(); in configure()
328 _col2im_kernel->set_target(CLScheduler::get().target()); in configure()
330 …_col2im_kernel->configure(compile_context, gemm_output_to_use, dst, Size2D(conv_w, conv_h), conv2d… in configure()
331 CLScheduler::get().tune_kernel_static(*_col2im_kernel.get()); in configure()
599 CLScheduler::get().enqueue_op(*_col2im_kernel.get(), pack, false); in run()
|
| D | ClGemmConv2d.h | 168 std::unique_ptr<opencl::kernels::ClCol2ImKernel> _col2im_kernel; variable
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuGemmConv2d.cpp | 95 …shape_kernel(nullptr), _im2col_kernel(), _mm_gemm(), _mm_gemmlowp(), _col2im_kernel(), _reshape_ke… in CpuGemmConv2d()
369 _col2im_kernel = std::make_unique<kernels::CpuCol2ImKernel>(); in configure()
370 _col2im_kernel->configure(gemm_output_to_use, dst, Size2D(conv_w, conv_h)); in configure()
621 …NEScheduler::get().schedule_op(_col2im_kernel.get(), Window::DimY, _col2im_kernel->window(), pack); in run()
|
| D | CpuGemmConv2d.h | 223 std::unique_ptr<kernels::CpuCol2ImKernel> _col2im_kernel; variable
|