Lines Matching refs:thread_pool_device

30   Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11));  in test_multithread_elementwise()
31 out.device(thread_pool_device) = in1 + in2 * 3.14f; in test_multithread_elementwise()
53 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11)); in test_multithread_compound_assignment()
54 out.device(thread_pool_device) = in1; in test_multithread_compound_assignment()
55 out.device(thread_pool_device) += in2 * 3.14f; in test_multithread_compound_assignment()
86 Eigen::ThreadPoolDevice thread_pool_device(&tp, 4); in test_multithread_contraction() local
89 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_multithread_contraction()
127 Eigen::ThreadPoolDevice thread_pool_device(&tp, 12); in test_contraction_corner_cases() local
130 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
145 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
162 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
180 t_result.device(thread_pool_device) = t_left.contract(t_right, dims); in test_contraction_corner_cases()
217 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_multithread_contraction_agrees_with_singlethread()
223 tp_result.device(thread_pool_device) = left.contract(right, dims); in test_multithread_contraction_agrees_with_singlethread()
256 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_full_contraction()
262 tp_result.device(thread_pool_device) = left.contract(right, dims); in test_full_contraction()
276 Eigen::ThreadPoolDevice thread_pool_device(&thread_pool, num_threads); in test_multithreaded_reductions() local
287 full_redux_tp.device(thread_pool_device) = t1.sum(); in test_multithreaded_reductions()
300 Eigen::ThreadPoolDevice thread_pool_device(&tp, num_threads); in test_memcpy() local
306 thread_pool_device.memcpy(&result[0], t1.data(), size*sizeof(float)); in test_memcpy()