Home
last modified time | relevance | path

Searched refs:ThreadPoolDevice (Results 1 – 9 of 9) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_thread_pool.cpp30 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11)); in test_multithread_elementwise()
53 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(3, 11)); in test_multithread_compound_assignment()
86 Eigen::ThreadPoolDevice thread_pool_device(&tp, 4); in test_multithread_contraction()
127 Eigen::ThreadPoolDevice thread_pool_device(&tp, 12); in test_contraction_corner_cases()
217 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_multithread_contraction_agrees_with_singlethread()
256 Eigen::ThreadPoolDevice thread_pool_device(&tp, internal::random<int>(2, 11)); in test_full_contraction()
276 Eigen::ThreadPoolDevice thread_pool_device(&thread_pool, num_threads); in test_multithreaded_reductions()
300 Eigen::ThreadPoolDevice thread_pool_device(&tp, num_threads); in test_memcpy()
317 Eigen::ThreadPoolDevice device(&tp, 2); in test_multithread_random()
330 Eigen::ThreadPoolDevice device(&threads, num_threads); in test_multithread_shuffle()
/external/eigen/bench/tensors/
Dcontraction_benchmarks_cpu.cc9 Eigen::ThreadPoolDevice device(&pool, threads);
18 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
Dtensor_benchmarks_cpu.cc9 Eigen::ThreadPoolDevice device(&pool, threads);
16 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
92 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, D1, D2, D3); \
141 BenchmarkSuite<Eigen::ThreadPoolDevice, float> suite(device, N); \
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorExecutor.h142 class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable> {
145 static inline void run(const Expression& expr, const ThreadPoolDevice& device)
147 typedef TensorEvaluator<Expression, ThreadPoolDevice> Evaluator;
162 num_threads = TensorCostModel<ThreadPoolDevice>::numThreads(
DTensorDeviceThreadPool.h107 struct ThreadPoolDevice { struct
109 …ThreadPoolDevice(ThreadPoolInterface* pool, int num_cores) : pool_(pool), num_threads_(num_cores) … in ThreadPoolDevice() function
185 typedef TensorCostModel<ThreadPoolDevice> CostModel; in parallelFor() argument
DTensorContractionThreadPool.h60 … TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, ThreadPoolDevice> :
61 …ensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, ThreadPoolDevice> > {
63 typedef ThreadPoolDevice Device;
205 int num_threads = TensorCostModel<ThreadPoolDevice>::numThreads(
705 double taskSize = TensorCostModel<ThreadPoolDevice>::taskSize(
DTensorForwardDeclarations.h72 struct ThreadPoolDevice;
DTensorReduction.h244 struct FullReducer<Self, Op, ThreadPoolDevice, Vectorizable> {
250 static void run(const Self& self, Op& reducer, const ThreadPoolDevice& device,
262 const int num_threads = TensorCostModel<ThreadPoolDevice>::numThreads(
DREADME.md420 presently three devices you can use: DefaultDevice, ThreadPoolDevice and
433 // Create the Eigen ThreadPoolDevice.
434 Eigen::ThreadPoolDevice my_device(4 /* number of threads to use */);