Lines Matching refs:m_functor

72   const FunctorType m_functor;  member in Kokkos::Impl::ParallelFor
82 m_functor(i); in exec_range()
89 m_functor(TagType(), i); in exec_range()
126 : m_functor(arg_functor), m_policy(arg_policy) {} in ParallelFor()
168 const FunctorType m_functor; member in Kokkos::Impl::ParallelReduce
195 m_functor(i, update); in exec_range()
202 m_functor(TagType(), i, update); in exec_range()
217 ReducerConditional::select(m_functor, m_reducer)) / in run()
222 ReducerConditional::select(m_functor, m_reducer), in run()
246 ReducerConditional::select(m_functor, m_reducer), blockIdx.x, in run()
263 ReducerConditional::select(m_functor, m_reducer), shared); in run()
279 ValueInit::init(ReducerConditional::select(m_functor, m_reducer), &value); in run()
303 ValueInit::init(ReducerConditional::select(m_functor, m_reducer), &init); in run()
306 ReducerConditional::select(m_functor, m_reducer), in run()
314 ValueJoin(ReducerConditional::select(m_functor, m_reducer)), in run()
320 ReducerConditional::select(m_functor, m_reducer), in run()
364 const int block_size = local_block_size(m_functor); in execute()
370 ReducerConditional::select(m_functor, m_reducer)) * in execute()
391 WorkTag>(m_functor, in execute()
405 ReducerConditional::select(m_functor, m_reducer)); in execute()
412 ValueInit::init(ReducerConditional::select(m_functor, m_reducer), in execute()
423 : m_functor(arg_functor), in ParallelReduce()
436 : m_functor(arg_functor), in ParallelReduce()
479 const FunctorType m_functor; member in Kokkos::Impl::ParallelScanHIPBase
492 m_functor(i, update, final_result); in exec_range()
500 m_functor(TagType(), i, update, final_result); in exec_range()
508 word_count(ValueTraits::value_size(m_functor) / sizeof(size_type)); in initial()
514 ValueInit::init(m_functor, shared_value); in initial()
534 m_functor, blockIdx.x, gridDim.x, in initial()
544 word_count(ValueTraits::value_size(m_functor) / sizeof(size_type)); in final()
563 ValueInit::init(m_functor, shared_accum); in final()
575 ValueInit::init(m_functor, shared_prefix + word_count.value); in final()
596 m_functor, in final()
651 const int block_size = static_cast<int>(local_block_size(m_functor)); in impl_execute()
668 ValueTraits::value_size(m_functor) * m_grid_x); in impl_execute()
674 const int shmem = ValueTraits::value_size(m_functor) * (block_size + 2); in impl_execute()
693 : m_functor(arg_functor), m_policy(arg_policy) {} in ParallelScanHIPBase()
728 const int size = Base::ValueTraits::value_size(Base::m_functor); in execute()