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()
134 const int intra_op_parallelism_threads) { in set_intra_op_parallelism_threads()137 int64 intra_op_parallelism_threads() const { in intra_op_parallelism_threads() function
87 HloRunner::HloRunner(se::Platform* platform, int intra_op_parallelism_threads) { in HloRunner()
119 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 intra_op_parallelism_threads = in EigenThreadPoolInfo() local
381 int32 intra_op_parallelism_threads = 2; field
1385 def intra_op_parallelism_threads(self): member in Context1389 def intra_op_parallelism_threads(self, num_threads): member in Context
835 const int64 intra_op_parallelism_threads = 24; in BM_ParallelFusion() local