Home
last modified time | relevance | path

Searched refs:intra_op_parallelism_threads (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlocal_device.cc76 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()
H A Dprocess_util.cc137 const int32 intra_op = options.config.intra_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbackend.cc54 int BackendOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::BackendOptions
94 computation_placer, options.intra_op_parallelism_threads())); in CreateBackend()
124 int intra_op_parallelism_threads) in Backend() argument
137 const int num_threads = intra_op_parallelism_threads > 0 in Backend()
138 ? intra_op_parallelism_threads in Backend()
H A Dhlo_module_config.h216 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads() argument
217 intra_op_parallelism_threads_ = intra_op_parallelism_threads; in set_intra_op_parallelism_threads()
219 int64_t intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
H A Dhlo_module_config.cc71 if (intra_op_parallelism_threads() > 0) { in compilation_cache_key()
73 intra_op_parallelism_threads()); in compilation_cache_key()
H A Dbackend.h54 int intra_op_parallelism_threads() const;
164 int intra_op_parallelism_threads);
H A Dhlo_runner.h54 int intra_op_parallelism_threads = -1);
H A Dhlo_runner.cc37 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner() argument
41 intra_op_parallelism_threads); in HloRunner()
H A Dlocal_service.cc56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
H A Dservice.h62 int intra_op_parallelism_threads() const;
H A Dservice.cc124 int ServiceOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::ServiceOptions
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient_library.cc31 int intra_op_parallelism_threads, in LocalClientOptions() argument
35 intra_op_parallelism_threads_(intra_op_parallelism_threads), in LocalClientOptions()
61 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::LocalClientOptions
112 options.intra_op_parallelism_threads()); in GetOrCreateLocalClient()
H A Dclient_library.h48 int intra_op_parallelism_threads = -1,
62 int intra_op_parallelism_threads() const;
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig.py104 return context.context().intra_op_parallelism_threads
118 context.context().intra_op_parallelism_threads = num_threads
H A Dconfig_test.py131 context.context().intra_op_parallelism_threads)
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Ddistribute_coordinator_test.py121 session_config.intra_op_parallelism_threads += 1
823 session_config.intra_op_parallelism_threads)
833 session_config.intra_op_parallelism_threads)
913 self.assertEqual(session_config.intra_op_parallelism_threads, 1)
923 session_config.intra_op_parallelism_threads = 1
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1076 config.intra_op_parallelism_threads = self._intra_op_parallelism_threads
1804 def intra_op_parallelism_threads(self): member in Context
1805 return self.config.intra_op_parallelism_threads
1807 @intra_op_parallelism_threads.setter
1808 def intra_op_parallelism_threads(self, num_threads): member in Context
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcpu_gpu_fusion_test.cc832 const int64_t intra_op_parallelism_threads = 24; in BM_ParallelFusion() local
835 client_options.set_intra_op_parallelism_threads(intra_op_parallelism_threads); in BM_ParallelFusion()
893 intra_op_parallelism_threads); in BM_ParallelFusion()
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-config-proto.pbtxt13 name: "intra_op_parallelism_threads"
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
H A Dcsr_sparse_matrix_ops_test.py1435 intra_op_parallelism_threads=num_threads)) as sess:
1493 intra_op_parallelism_threads=num_threads)) as sess:
1535 intra_op_parallelism_threads=num_threads)) as sess:
1608 intra_op_parallelism_threads=num_threads)) as sess:
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc216 session_options.config.intra_op_parallelism_threads()), in XlaDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc673 module->config().intra_op_parallelism_threads() > 0 in RunHloPassesAfterLayoutAssn()
674 ? module->config().intra_op_parallelism_threads() in RunHloPassesAfterLayoutAssn()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dreader_ops_test.py713 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto439 // parallelized on a pool of intra_op_parallelism_threads.
452 int32 intra_op_parallelism_threads = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dimage_ops_test.py578 config.intra_op_parallelism_threads = cpu_count
608 config.intra_op_parallelism_threads = cpu_count
638 config.intra_op_parallelism_threads = cpu_count
699 config.intra_op_parallelism_threads = cpu_count
742 config.intra_op_parallelism_threads = cpu_count

12