Searched refs:prev_num_threads (Results 1 – 3 of 3) sorted by relevance
81 int prev_num_threads = mkl_set_num_threads_local( in __xla_cpu_runtime_MKLMatMulF32() local85 mkl_set_num_threads_local(prev_num_threads); in __xla_cpu_runtime_MKLMatMulF32()97 int prev_num_threads = mkl_set_num_threads_local( in __xla_cpu_runtime_MKLMatMulF64() local101 mkl_set_num_threads_local(prev_num_threads); in __xla_cpu_runtime_MKLMatMulF64()111 int prev_num_threads = mkl_set_num_threads_local(1); in __xla_cpu_runtime_MKLSingleThreadedMatMulF32() local114 mkl_set_num_threads_local(prev_num_threads); in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()124 int prev_num_threads = mkl_set_num_threads_local(1); in __xla_cpu_runtime_MKLSingleThreadedMatMulF64() local127 mkl_set_num_threads_local(prev_num_threads); in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
572 let prev_num_threads = self in set_num_threads() localVariable576 if let Some(num_spawn) = num_threads.checked_sub(prev_num_threads) { in set_num_threads()
1316 prev_num_threads = torch.get_num_threads()1325 torch.set_num_threads(prev_num_threads)