| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | local_device.cc | 76 int32 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::NumSchedulableCPUs(); in EigenThreadPoolInfo()
89 intra_op_parallelism_threads /= port::NUMANumNodes(); in EigenThreadPoolInfo()
95 eigen_worker_threads_.num_threads = intra_op_parallelism_threads; in EigenThreadPoolInfo()
98 intra_op_parallelism_threads); in EigenThreadPoolInfo()
|
| /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
117 computation_placer, options.intra_op_parallelism_threads())); in CreateBackend()
147 int intra_op_parallelism_threads) in Backend() argument
165 const int num_threads = intra_op_parallelism_threads > 0 in Backend()
166 ? intra_op_parallelism_threads in Backend()
|
| D | hlo_module_config.h | 98 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads() argument
99 intra_op_parallelism_threads_ = intra_op_parallelism_threads; in set_intra_op_parallelism_threads()
101 int64 intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
|
| D | hlo_module_config.cc | 60 if (intra_op_parallelism_threads() > 0) { in compilation_cache_key()
62 intra_op_parallelism_threads()); in compilation_cache_key()
|
| D | backend.h | 56 int intra_op_parallelism_threads() const;
163 int intra_op_parallelism_threads);
|
| D | local_service.cc | 56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
|
| D | service.h | 63 int intra_op_parallelism_threads() const;
|
| /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 | 82 return context.context().intra_op_parallelism_threads
96 context.context().intra_op_parallelism_threads = num_threads
|
| D | config_test.py | 153 context.context().intra_op_parallelism_threads)
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 758 def intra_op_parallelism_threads(self): member in 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 Context
767 self._config.intra_op_parallelism_threads = num_threads
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.threading.pbtxt | 8 name: "intra_op_parallelism_threads"
|
| D | tensorflow.-config-proto.pbtxt | 13 name: "intra_op_parallelism_threads"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.threading.pbtxt | 8 name: "intra_op_parallelism_threads"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_coordinator_test.py | 129 session_config.intra_op_parallelism_threads += 1
829 session_config.intra_op_parallelism_threads)
839 session_config.intra_op_parallelism_threads)
919 self.assertEqual(session_config.intra_op_parallelism_threads, 1)
929 session_config.intra_op_parallelism_threads = 1
|
| /external/tensorflow/tensorflow/contrib/image/python/kernel_tests/ |
| D | distort_image_ops_test.py | 263 config.intra_op_parallelism_threads = cpu_count
307 config.intra_op_parallelism_threads = cpu_count
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/ |
| D | sdca_estimator_test.py | 38 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
325 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | run_config.py | 313 intra_op_parallelism_threads=num_cores,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 835 const int64 intra_op_parallelism_threads = 24; in BM_ParallelFusion() local
838 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 | 369 module->config().intra_op_parallelism_threads() > 0 in RunHloPassesAfterLayoutAssn()
370 ? module->config().intra_op_parallelism_threads() in RunHloPassesAfterLayoutAssn()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | config.proto | 323 // parallelized on a pool of intra_op_parallelism_threads.
325 int32 intra_op_parallelism_threads = 2; field
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | experiment_test.py | 313 expected_config_proto.intra_op_parallelism_threads = 15
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_test.py | 490 config.intra_op_parallelism_threads = cpu_count
520 config.intra_op_parallelism_threads = cpu_count
550 config.intra_op_parallelism_threads = cpu_count
611 config.intra_op_parallelism_threads = cpu_count
654 config.intra_op_parallelism_threads = cpu_count
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reader_ops_test.py | 716 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
|