Searched refs:gather_kernel (Results 1 – 5 of 5) sorted by relevance
30 class gather_kernel : public meta_kernel class33 gather_kernel() : meta_kernel("gather") in gather_kernel() function in boost::compute::detail::gather_kernel82 detail::gather_kernel<InputIterator, MapIterator, OutputIterator> kernel; in gather()
96 auto gather_kernel = reinterpret_cast<GatherInt8CPUKernel *>(cdata); in GatherInt8Run() local97 auto error_code = gather_kernel->DoGather(task_id); in GatherInt8Run()
85 auto gather_kernel = reinterpret_cast<GatherCPUKernel *>(cdata); in GatherRun() local86 auto error_code = gather_kernel->DoGather(task_id); in GatherRun()
111 auto gather_kernel = reinterpret_cast<GatherFp16CPUKernel *>(cdata); in GatherRunFp16() local112 auto error_code = gather_kernel->DoGather(task_id); in GatherRunFp16()