Searched refs:intra_op_thread_pool (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d.cc | 40 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF32()
42 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF32()
64 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF16()
66 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_fft.cc | 35 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenFft()
37 *run_options->intra_op_thread_pool(), out, operand, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fork_join.cc | 77 CHECK_NE(run_options->intra_op_thread_pool(), nullptr); in __xla_cpu_runtime_ParallelForkJoin()
88 run_options->intra_op_thread_pool()->enqueueNoNotification( in __xla_cpu_runtime_ParallelForkJoin()
|
| D | runtime_matmul_mkl.cc | 86 run_options->intra_op_thread_pool()->numThreads()); in __xla_cpu_runtime_MKLMatMulF32()
101 run_options->intra_op_thread_pool()->numThreads()); in __xla_cpu_runtime_MKLMatMulF64()
|
| D | runtime_matmul.cc | 70 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in MatMul()
71 C.device(*run_options->intra_op_thread_pool()) = A.contract(B, dims); in MatMul()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 75 const Eigen::ThreadPoolDevice* intra_op_thread_pool) { in set_intra_op_thread_pool() argument
76 intra_op_thread_pool_ = intra_op_thread_pool; in set_intra_op_thread_pool()
80 const Eigen::ThreadPoolDevice* ExecutableRunOptions::intra_op_thread_pool() in intra_op_thread_pool() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 119 const Eigen::ThreadPoolDevice* intra_op_thread_pool);
120 const Eigen::ThreadPoolDevice* intra_op_thread_pool() const;
|