HomeSort by relevance Sort by last modified time
    Searched refs:intra_op_thread_pool (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 59 const Eigen::ThreadPoolDevice* intra_op_thread_pool) {
60 intra_op_thread_pool_ = intra_op_thread_pool;
64 const Eigen::ThreadPoolDevice* ExecutableRunOptions::intra_op_thread_pool() function in class:xla::ExecutableRunOptions
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/
runtime_fft.cc 34 tensorflow::xla::EigenFftImpl(*run_options->intra_op_thread_pool(), out,
runtime_fork_join.cc 77 CHECK_NE(run_options->intra_op_thread_pool(), nullptr);
88 run_options->intra_op_thread_pool()->enqueueNoNotification(
runtime_conv2d.cc 38 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch,
57 *run_options->intra_op_thread_pool(), out, lhs, rhs, input_batch,
runtime_matmul_mkl.cc 84 // number specified in intra_op_thread_pool to MKL.
86 run_options->intra_op_thread_pool()->numThreads());
99 // number specified in intra_op_thread_pool to MKL.
101 run_options->intra_op_thread_pool()->numThreads());
runtime_matmul.cc 72 C.device(*run_options->intra_op_thread_pool()) = A.contract(B, dims);

Completed in 115 milliseconds