/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A 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()
|
H A D | process_util.cc | 137 const int32 intra_op = options.config.intra_op_parallelism_threads(); in NumInterOpThreadsFromSessionOptions()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | backend.cc | 54 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 D | hlo_module_config.h | 216 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 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()
|
H A D | backend.h | 54 int intra_op_parallelism_threads() const; 164 int intra_op_parallelism_threads);
|
H A D | hlo_runner.h | 54 int intra_op_parallelism_threads = -1);
|
H A 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()
|
H A D | local_service.cc | 56 .set_intra_op_parallelism_threads(options.intra_op_parallelism_threads()) in NewService()
|
H A D | service.h | 62 int intra_op_parallelism_threads() const;
|
H A D | service.cc | 124 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 D | client_library.cc | 31 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 D | client_library.h | 48 int intra_op_parallelism_threads = -1, 62 int intra_op_parallelism_threads() const;
|
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
H A D | config.py | 104 return context.context().intra_op_parallelism_threads 118 context.context().intra_op_parallelism_threads = num_threads
|
H A D | config_test.py | 131 context.context().intra_op_parallelism_threads)
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | distribute_coordinator_test.py | 121 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 D | context.py | 1076 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 D | cpu_gpu_fusion_test.cc | 832 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 D | tensorflow.-config-proto.pbtxt | 13 name: "intra_op_parallelism_threads"
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
H A D | csr_sparse_matrix_ops_test.py | 1435 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 D | xla_device.cc | 216 session_options.config.intra_op_parallelism_threads()), in XlaDevice()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_compiler.cc | 673 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 D | reader_ops_test.py | 713 inter_op_parallelism_threads=1, intra_op_parallelism_threads=1)
|
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
H A D | config.proto | 439 // 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 D | image_ops_test.py | 578 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
|