Searched refs:intra_op_thread_pool (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 59 const Eigen::ThreadPoolDevice* intra_op_thread_pool) { in set_intra_op_thread_pool() argument
60 intra_op_thread_pool_ = intra_op_thread_pool; in set_intra_op_thread_pool()
64 const Eigen::ThreadPoolDevice* ExecutableRunOptions::intra_op_thread_pool() in intra_op_thread_pool() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 69 const Eigen::ThreadPoolDevice* intra_op_thread_pool);
70 const Eigen::ThreadPoolDevice* intra_op_thread_pool() const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| 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_conv2d.cc | 38 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF32()
57 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch, in __xla_cpu_runtime_EigenConvF16()
|
| 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_fft.cc | 34 tensorflow::xla::EigenFftImpl(*run_options->intra_op_thread_pool(), out, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_matmul.cc | 72 C.device(*run_options->intra_op_thread_pool()) = A.contract(B, dims); in MatMul()
|