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

  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 58 ExecutableRunOptions& ExecutableRunOptions::set_intra_op_thread_pool( function in class:xla::ExecutableRunOptions
executable_run_options.h 68 ExecutableRunOptions& set_intra_op_thread_pool(
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiled_cpu_function.h 125 run_options_.set_intra_op_thread_pool(pool);
  /external/tensorflow/tensorflow/compiler/xla/tests/
multiple_devices_on_host_test.cc 41 execute_options.set_intra_op_thread_pool(
local_client_test_base.cc 152 run_options.set_intra_op_thread_pool(thread_pool_wrapper_->device.get());
xla_hlo_profile_test.cc 173 exec_run_options.set_intra_op_thread_pool(
exhaustive_op_test.cc 305 run_opts.set_intra_op_thread_pool(
fusion_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/jit/
xla_compile_on_demand_op.cc 78 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device());
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_runtime_test.cc 107 run_options.set_intra_op_thread_pool(&device);
  /external/tensorflow/tensorflow/compiler/jit/kernels/
xla_ops.cc 363 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device());
537 run_options.set_intra_op_thread_pool(&ctx->eigen_cpu_device());
  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_execute_op.cc 210 run_options.set_intra_op_thread_pool(&context->eigen_cpu_device());
  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 248 options.set_intra_op_thread_pool(
296 options.set_intra_op_thread_pool(
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_runner.cc 448 run_options.set_intra_op_thread_pool(
service.cc 451 options.set_intra_op_thread_pool(
546 options.set_intra_op_thread_pool(
    [all...]

Completed in 928 milliseconds