Searched defs:intra_op_parallelism_threads (Results 1 – 9 of 9) sorted by relevance
55 int BackendOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::BackendOptions125 int intra_op_parallelism_threads) in Backend()
177 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads()180 int64 intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
37 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner()
122 int ServiceOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::ServiceOptions
29 int intra_op_parallelism_threads, in LocalClientOptions()59 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function in xla::LocalClientOptions
76 int32_t intra_op_parallelism_threads = in EigenThreadPoolInfo() local
420 int32 intra_op_parallelism_threads = 2; field
1705 def intra_op_parallelism_threads(self): member in Context1709 def intra_op_parallelism_threads(self, num_threads): member in Context
834 const int64_t intra_op_parallelism_threads = 24; in BM_ParallelFusion() local