Searched refs:AsEigenThreadPool (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base_test.cc | 31 Eigen::ThreadPoolDevice eigen_device(pool.AsEigenThreadPool(), in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 51 eigen_worker_threads_.workers->AsEigenThreadPool(), in SingleThreadedCpuDevice()
|
| D | local_device.cc | 96 eigen_worker_threads_.workers->AsEigenThreadPool(); in EigenThreadPoolInfo()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 39 eigen_worker_threads_.workers->AsEigenThreadPool(), in DeviceSimple()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantization_utils_test.cc | 215 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestRequantizeManyInNewRangeEigenVsNonEigen()
297 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TimeRequantizeManyInNewRange()
609 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestRequantizeManyInNewRange32To8BitUsingEigen()
640 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestFloatToQuantizedInPlaceUsingEigen()
651 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestOverflowWithEigen()
719 Eigen::ThreadPoolDevice eigen_device(threadpool.AsEigenThreadPool(), in TestQuantizedToFloatInPlaceUsingEigen()
|
| D | nn_ops_test.cc | 764 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_LRNFloat()
845 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPool()
935 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_AvgPoolBk()
1040 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_MaxPool()
1219 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_ReluFloat()
1290 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), in BM_SoftplusFloat()
|
| D | quantized_batch_norm_op_test.cc | 220 Eigen::ThreadPoolDevice eigen_cpu_device(threadpool.AsEigenThreadPool(), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_threadpool.h | 67 ->workers->AsEigenThreadPool()) {} in MklDnnThreadPool()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | threadpool.h | 216 Eigen::ThreadPoolInterface* AsEigenThreadPool() const;
|
| D | threadpool.cc | 266 Eigen::ThreadPoolInterface* ThreadPool::AsEigenThreadPool() const { in AsEigenThreadPool() function in tensorflow::thread::ThreadPool
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 75 device(new Eigen::ThreadPoolDevice(pool->AsEigenThreadPool(), in IntraOpThreadPool()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 117 device(new Eigen::ThreadPoolDevice(pool->AsEigenThreadPool(), in EigenThreadPoolWrapper()
|
| D | cpu_gpu_fusion_test.cc | 899 Eigen::ThreadPoolDevice device(pool.AsEigenThreadPool(), pool.NumThreads()); in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime_test.cc | 103 Eigen::ThreadPoolDevice device(pool.AsEigenThreadPool(), pool.NumThreads()); in EigenMatrixMultiply()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 322 Eigen::ThreadPoolDevice thread_pool(pool.AsEigenThreadPool(), in ReplayComputation()
|