| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | local_device.cc | 76 int32_t intra_op_parallelism_threads = in EigenThreadPoolInfo() local
77 options.config.intra_op_parallelism_threads(); in EigenThreadPoolInfo()
79 if (intra_op_parallelism_threads == 0) { in EigenThreadPoolInfo()
81 intra_op_parallelism_threads = env_num_threads; in EigenThreadPoolInfo()
83 if (intra_op_parallelism_threads == 0) { in EigenThreadPoolInfo()
84 intra_op_parallelism_threads = port::MaxParallelism(numa_node); in EigenThreadPoolInfo()
89 eigen_worker_threads_.num_threads = intra_op_parallelism_threads; in EigenThreadPoolInfo()
92 intra_op_parallelism_threads, in EigenThreadPoolInfo()
|
| D | process_util.cc | 137 const int32 intra_op = options.config.intra_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 55 int BackendOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::BackendOptions
95 computation_placer, options.intra_op_parallelism_threads())); in CreateBackend()
125 int intra_op_parallelism_threads) in Backend() argument
138 const int num_threads = intra_op_parallelism_threads > 0 in Backend()
139 ? intra_op_parallelism_threads in Backend()
|
| D | hlo_module_config.cc | 71 if (intra_op_parallelism_threads() > 0) { in compilation_cache_key()
73 intra_op_parallelism_threads()); in compilation_cache_key()
|
| D | hlo_module_config.h | 177 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads() argument
178 intra_op_parallelism_threads_ = intra_op_parallelism_threads; in set_intra_op_parallelism_threads()
180 int64 intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
|
| D | backend.h | 56 int intra_op_parallelism_threads() const;
166 int intra_op_parallelism_threads);
|
| D | hlo_runner.h | 54 int intra_op_parallelism_threads = -1);
|
| D | local_service.cc | 57 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
|
| D | service.h | 63 int intra_op_parallelism_threads() const;
|
| D | hlo_runner.cc | 37 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner() argument
41 intra_op_parallelism_threads); in HloRunner()
|
| D | service.cc | 122 int ServiceOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::ServiceOptions
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 29 int intra_op_parallelism_threads, in LocalClientOptions() argument
33 intra_op_parallelism_threads_(intra_op_parallelism_threads), in LocalClientOptions()
59 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::LocalClientOptions
110 options.intra_op_parallelism_threads()); in GetOrCreateLocalClient()
|
| D | client_library.h | 50 int intra_op_parallelism_threads = -1,
64 int intra_op_parallelism_threads() const;
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 108 return context.context().intra_op_parallelism_threads
122 context.context().intra_op_parallelism_threads = num_threads
|
| D | config_test.py | 132 context.context().intra_op_parallelism_threads)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator_test.py | 132 session_config.intra_op_parallelism_threads += 1
837 session_config.intra_op_parallelism_threads)
847 session_config.intra_op_parallelism_threads)
927 self.assertEqual(session_config.intra_op_parallelism_threads, 1)
937 session_config.intra_op_parallelism_threads = 1
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1015 config.intra_op_parallelism_threads = self._intra_op_parallelism_threads
1705 def intra_op_parallelism_threads(self): member in Context
1706 return self.config.intra_op_parallelism_threads
1708 @intra_op_parallelism_threads.setter
1709 def intra_op_parallelism_threads(self, num_threads): member in Context
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-config-proto.pbtxt | 13 name: "intra_op_parallelism_threads"
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 834 const int64_t intra_op_parallelism_threads = 24; in BM_ParallelFusion() local
837 client_options.set_intra_op_parallelism_threads(intra_op_parallelism_threads); in BM_ParallelFusion()
898 intra_op_parallelism_threads); in BM_ParallelFusion()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 456 module->config().intra_op_parallelism_threads() > 0 in RunHloPassesAfterLayoutAssn()
457 ? module->config().intra_op_parallelism_threads() in RunHloPassesAfterLayoutAssn()
|
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
| D | csr_sparse_matrix_ops_test.py | 1420 intra_op_parallelism_threads=num_threads)) as sess:
1478 intra_op_parallelism_threads=num_threads)) as sess:
1520 intra_op_parallelism_threads=num_threads)) as sess:
1593 intra_op_parallelism_threads=num_threads)) as sess:
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 216 session_options.config.intra_op_parallelism_threads()), in XlaDevice()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reader_ops_test.py | 717 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 407 // parallelized on a pool of intra_op_parallelism_threads.
420 int32 intra_op_parallelism_threads = 2; field
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 550 config.intra_op_parallelism_threads = cpu_count
580 config.intra_op_parallelism_threads = cpu_count
610 config.intra_op_parallelism_threads = cpu_count
671 config.intra_op_parallelism_threads = cpu_count
714 config.intra_op_parallelism_threads = cpu_count
|