Lines Matching refs:m_functor

75   const FunctorType m_functor;  member in Kokkos::Impl::ParallelFor
119 ParallelFor::template exec_range<WorkTag>(self.m_functor, range.begin(), in exec_schedule()
149 ParallelFor::template exec_range<WorkTag>(self.m_functor, begin, end); in exec_schedule()
163 : m_functor(arg_functor), m_policy(arg_policy) {} in ParallelFor()
182 const FunctorType m_functor; member in Kokkos::Impl::ParallelFor
211 ParallelFor::exec_range(self.m_mdr_policy, self.m_functor, range.begin(), in exec_schedule()
239 ParallelFor::exec_range(self.m_mdr_policy, self.m_functor, begin, end); in exec_schedule()
253 : m_functor(arg_functor), in ParallelFor()
270 const FunctorType m_functor; member in Kokkos::Impl::ParallelFor
320 self.m_functor, Member(&exec, self.m_policy, self.m_shared)); in exec()
332 policy.team_size_recommended(m_functor, ParallelForTag{})); in fix_policy()
348 : m_functor(arg_functor), in ParallelFor()
385 const FunctorType m_functor; member in Kokkos::Impl::ParallelReduce
431 self.m_functor, range.begin(), range.end(), in exec_schedule()
433 ReducerConditional::select(self.m_functor, self.m_reducer), in exec_schedule()
437 ReducerConditional::select(self.m_functor, self.m_reducer)); in exec_schedule()
455 ReducerConditional::select(self.m_functor, self.m_reducer), in exec_schedule()
465 ParallelReduce::template exec_range<WorkTag>(self.m_functor, begin, end, in exec_schedule()
471 ReducerConditional::select(self.m_functor, self.m_reducer)); in exec_schedule()
478 ValueInit::init(ReducerConditional::select(m_functor, m_reducer), in execute()
480 ValueFinal::final(ReducerConditional::select(m_functor, m_reducer), in execute()
486 ReducerConditional::select(m_functor, m_reducer)), in execute()
498 ReducerConditional::select(m_functor, m_reducer)); in execute()
513 : m_functor(arg_functor), in ParallelReduce()
527 : m_functor(arg_functor), in ParallelReduce()
570 const FunctorType m_functor; member in Kokkos::Impl::ParallelReduce
601 self.m_mdr_policy, self.m_functor, range.begin(), range.end(), in exec_schedule()
603 ReducerConditional::select(self.m_functor, self.m_reducer), in exec_schedule()
607 ReducerConditional::select(self.m_functor, self.m_reducer)); in exec_schedule()
623 ReducerConditional::select(self.m_functor, self.m_reducer), in exec_schedule()
632 ParallelReduce::exec_range(self.m_mdr_policy, self.m_functor, begin, end, in exec_schedule()
638 ReducerConditional::select(self.m_functor, self.m_reducer)); in exec_schedule()
645 ReducerConditional::select(m_functor, m_reducer)), in execute()
657 ReducerConditional::select(m_functor, m_reducer)); in execute()
671 : m_functor(arg_functor), in ParallelReduce()
686 : m_functor(arg_functor), in ParallelReduce()
726 const FunctorType m_functor; member in Kokkos::Impl::ParallelReduce
757 self.m_functor, Member(&exec, self.m_policy, self.m_shared), in exec()
759 ReducerConditional::select(self.m_functor, self.m_reducer), in exec()
763 ReducerConditional::select(self.m_functor, self.m_reducer)); in exec()
770 ValueInit::init(ReducerConditional::select(m_functor, m_reducer), in execute()
772 ValueFinal::final(ReducerConditional::select(m_functor, m_reducer), in execute()
778 ReducerConditional::select(m_functor, m_reducer)), in execute()
790 ReducerConditional::select(m_functor, m_reducer)); in execute()
805 m_functor, m_reducer, ParallelReduceTag{})); in fix_policy()
817 : m_functor(arg_functor), in ParallelReduce()
827 : m_functor(arg_functor), in ParallelReduce()
859 const FunctorType m_functor; member in Kokkos::Impl::ParallelScan
897 ValueInit::init(self.m_functor, exec.reduce_memory()); in exec()
899 ParallelScan::template exec_range<WorkTag>(self.m_functor, range.begin(), in exec()
903 exec.template scan_small<FunctorType, WorkTag>(self.m_functor); in exec()
905 ParallelScan::template exec_range<WorkTag>(self.m_functor, range.begin(), in exec()
913 ThreadsExec::resize_scratch(2 * ValueTraits::value_size(m_functor), 0); in execute()
919 : m_functor(arg_functor), m_policy(arg_policy) {} in ParallelScan()
936 const FunctorType m_functor; member in Kokkos::Impl::ParallelScanWithTotal
975 ValueInit::init(self.m_functor, exec.reduce_memory()); in exec()
978 self.m_functor, range.begin(), range.end(), update, false); in exec()
981 exec.template scan_small<FunctorType, WorkTag>(self.m_functor); in exec()
984 self.m_functor, range.begin(), range.end(), update, true); in exec()
995 ThreadsExec::resize_scratch(2 * ValueTraits::value_size(m_functor), 0); in execute()
1002 : m_functor(arg_functor), in ParallelScanWithTotal()