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

  /external/tensorflow/tensorflow/compiler/xla/client/
client_library.cc 29 int intra_op_parallelism_threads,
33 intra_op_parallelism_threads_(intra_op_parallelism_threads),
59 int LocalClientOptions::intra_op_parallelism_threads() const { function in class:xla::LocalClientOptions
110 options.intra_op_parallelism_threads());
  /external/tensorflow/tensorflow/compiler/xla/service/
backend.cc 55 int BackendOptions::intra_op_parallelism_threads() const { function in class:xla::BackendOptions
117 computation_placer, options.intra_op_parallelism_threads()));
147 int intra_op_parallelism_threads)
165 const int num_threads = intra_op_parallelism_threads > 0
166 ? intra_op_parallelism_threads
hlo_module_config.h 98 const int intra_op_parallelism_threads) {
99 intra_op_parallelism_threads_ = intra_op_parallelism_threads;
101 int64 intra_op_parallelism_threads() const { function in class:xla::HloModuleConfig
service.cc 119 int ServiceOptions::intra_op_parallelism_threads() const { function in class:xla::ServiceOptions
    [all...]
  /external/tensorflow/tensorflow/core/common_runtime/
local_device.cc 76 int32 intra_op_parallelism_threads = local
77 options.config.intra_op_parallelism_threads();
79 if (intra_op_parallelism_threads == 0) {
81 intra_op_parallelism_threads = env_num_threads;
83 if (intra_op_parallelism_threads == 0) {
84 intra_op_parallelism_threads = port::NumSchedulableCPUs();
89 intra_op_parallelism_threads /= port::NUMANumNodes();
95 eigen_worker_threads_.num_threads = intra_op_parallelism_threads;
98 intra_op_parallelism_threads);
  /external/tensorflow/tensorflow/compiler/xla/tests/
fusion_test.cc 835 const int64 intra_op_parallelism_threads = 24; local
898 intra_op_parallelism_threads); local
    [all...]
  /external/tensorflow/tensorflow/python/eager/
context.py 758 def intra_op_parallelism_threads(self): member in class:Context
759 return self._config.intra_op_parallelism_threads
761 @intra_op_parallelism_threads.setter
762 def intra_op_parallelism_threads(self, num_threads): member in class:Context
767 self._config.intra_op_parallelism_threads = num_threads
    [all...]

Completed in 2036 milliseconds